如何在熊猫中使用groupby计算最近14天的平均值

时间:2020-05-19 09:18:41

标签: python pandas pandas-groupby

我希望能够计算出菲律宾每个城市的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        |

但是我碰到了初学者。我敢肯定解决方案是如此简单。 :)

帮助。

谢谢。

1 个答案:

答案 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()