我希望能够计算出菲律宾每个城市的Covid-19病例的增长率。我使用的是卫生部门的公共病例数据集,其中包含caseID,病例确认的日期,病例所在的城市,等等。
我尝试使用groupby方法将数据划分为城市和日期,每天对案件进行计数,计算这些计数的累计总和,每日百分比变化以及最后14天的百分比变化平均值。 >
我能够做到,但我不知道如何将它们重新组合到包含城市及其过去14天平均增长率的数据集中。
这是我到目前为止所拥有的:
import pandas as pd
cases_df = pd.read_csv('https://storage.googleapis.com/doh_datadrops/DOH%20Data%20Drop%2020200518.csv',
parse_dates=[5,6,7,9,17])
#cast data to appropriate types for easy handling
for col in ['AgeGroup', 'Sex','RemovalType', 'Admitted', 'RegionRes','ProvRes','CityMunRes',
'CityMuniPSGC','HealthStatus','Quarantined','Pregnanttab']:
cases_df[col] = cases_df[col].astype('category')
for col in ['DateRepConf', 'DateDied', 'DateRecover', 'DateRepRem','DateOnset']:
cases_df[col] = cases_df[col].astype('datetime64')
cases_df.Age = cases_df.Age.astype('Int64')
list_cases_by_city_by_date = cases_df.groupby(['CityMunRes','DateRepConf'])['CaseCode'].count().cumsum().pct_change()
最后一个groupby产生一个多索引列表:
CityMunRes DateRepConf
ABUCAY 2020-01-30 NaN
2020-02-03 NaN
2020-02-05 NaN
2020-03-06 NaN
2020-03-07 NaN
...
ZARRAGA 2020-05-14 0.0
2020-05-15 0.0
2020-05-16 0.0
2020-05-17 0.0
2020-05-18 0.0
Name: CaseCode, Length: 28336, dtype: float64
现在,如果我指定城市,则可以使用以下代码计算最近14天的增长率:
list_cases_by_city_by_date.loc['ABUCAY'].tail(14).mean()
0.03571428571428571
但是我想要的是一个简单的数据框,其中包含“ CityMunRes”和“ Ave_Growth”作为列,如下所示:
| | City | Ave_Growth |
|---|--------|------------|
| 0 | ABUCAY | 0.03 |
| 1 | Bxxxx | 0.02 |
| 2 | Cxxxx | 0.50 |
| | ... | |
| n | Zxxx | 0.2 |
但是我碰到了初学者。我敢肯定解决方案是如此简单。 :)
帮助。
谢谢。
答案 0 :(得分:0)
编辑:匹配解决方案:
list_cases_by_city_by_date.groupby(level=0).apply(lambda x: x.tail(14).mean())
这个怎么样?您使用自己的一些代码制作一个字典,然后将它们全部放入一个数据帧中。
d = {city: list_cases_by_city_by_date.loc[city].tail(14).mean()
for city in list_cases_by_city_by_date.index.get_level_values('CityMunRes')}
pd.DataFrame(data=d.values(), index=d.keys(), columns=['mean'])
编辑:顺便说一句,当我尝试运行您提供的代码时,我得到0.75
,而不是0.03571428571428571
list_cases_by_city_by_date.loc['ABUCAY'].tail(14).mean()