From f91a74c7aa31a742e50629e4e57736cf5192320d Mon Sep 17 00:00:00 2001 From: Christian Heller Date: Sat, 4 Jul 2020 02:02:48 +0200 Subject: [PATCH] Move sum_district to the left, as it's of more interest than others. --- enhance_table.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/enhance_table.py b/enhance_table.py index 2410099..4ce7792 100755 --- a/enhance_table.py +++ b/enhance_table.py @@ -94,6 +94,10 @@ for line in lines[1:]: db[district][date] = {'new_infections': int(district_data)} sorted_dates.sort() +# Define and move sum_district from end to start. +sum_district = sorted_districts.pop() +sorted_districts.insert(0, sum_district) + # In LaGeSo's data, the last "district" is actually the sum of all districts / # the whole of Berlin. # @@ -103,9 +107,8 @@ sorted_dates.sort() # the table file, since we should have a human look at what mistake was # probably made. for date in sorted_dates: - sum_district = sorted_districts[-1] day_sum = 0 - for district in sorted_districts[:-1]: + for district in [d for d in sorted_districts if not d==sum_district]: day_sum += db[district][date]['new_infections'] if day_sum != db[sum_district][date]['new_infections']: raise Exception('Questionable district infection sum in %s' % date) -- 2.30.2