From: Christian Heller Date: Sat, 4 Jul 2020 00:02:48 +0000 (+0200) Subject: Move sum_district to the left, as it's of more interest than others. X-Git-Url: https://plomlompom.com/repos/todos?a=commitdiff_plain;h=f91a74c7aa31a742e50629e4e57736cf5192320d;p=berlin-corona-table Move sum_district to the left, as it's of more interest than others. --- 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)