X-Git-Url: https://plomlompom.com/repos/unset_cookie?a=blobdiff_plain;f=enhance_table.py;h=4ce77925c91c985d1bfb177cb0d7ac5694d2d690;hb=f91a74c7aa31a742e50629e4e57736cf5192320d;hp=cfb5eff9dba7e1fde9cb7709d2f00cc9f4c3cc51;hpb=c0b58478e632295d51a5a8dc6cee298c22b0b98c;p=berlin-corona-table diff --git a/enhance_table.py b/enhance_table.py index cfb5eff..4ce7792 100755 --- a/enhance_table.py +++ b/enhance_table.py @@ -32,7 +32,7 @@ translate = { 'TS': 'Tempelhof-Schöneberg', 'TK': 'Treptow-Köpenick', 'sum': 'all of Berlin', - 'wsum': 'sum for last 7 days', + 'wsum': 'sum of new infections for last 7 days', 'wavg': 'per-day average of new infections for last 7 days', 'winc': 'incidence (x per 100k inhabitants) of new infections for last 7 days', } @@ -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) @@ -150,7 +153,6 @@ th { text-align: left; vertical-align: bottom; }