使用pandas的一周中每天的平均操作数

时间:2017-08-28 15:16:40

标签: python pandas dataframe

假设我计算每小时的事件数量如下:

np.random.seed(42)
idx = pd.date_range('2017-01-01', '2017-01-14', freq='1H')
df = pd.DataFrame(np.random.choice([1,2,3,4,5,6], size=idx.shape[0]), index=idx, columns=['count'])
df.head()

Out[3]:
                     count
2017-01-01 00:00:00      4
2017-01-01 01:00:00      5
2017-01-01 02:00:00      3
2017-01-01 03:00:00      5
2017-01-01 04:00:00      5

如果我想知道一周中每天的次数,我可以这样做:

df.pivot_table(values='count', index=df.index.dayofweek, aggfunc='sum')

df.groupby(df.index.dayofweek).sum()

两种产量:

Out[4]:
   count
0    161
1    170
2    164
3    133
4    169
5     98
6    172

但是,如果我想计算每个工作日的平均事件数,则以下

df.pivot_table(values='count', index=df.index.dayofweek, aggfunc='mean') # [#1]

错误 !! 此方法使用总和(如上所述计算),并将其除以一周中每天出现的小时数。< / p>

我找到的解决方法是:

df_by_day = df.resample('1d').sum()
df_by_day.pivot_table(values='count', index=df_by_day.index.dayofweek, aggfunc='mean')

即,首先重新采样到几天,然后转动它。不知怎的,[#1]中的方法对我来说很自然。是否有更多的pythonic方式来实现我想要的?为什么没有重新采样平均值是错误的计算?

1 个答案:

答案 0 :(得分:3)

Resample first using df.resample and then df.groupby

df = df.resample('1d').sum()
print(df)

            count
2017-01-01     92
2017-01-02     86
2017-01-03     86
2017-01-04     90
2017-01-05     64
2017-01-06     82
2017-01-07     97
2017-01-08     80
2017-01-09     75
2017-01-10     84
2017-01-11     74
2017-01-12     69
2017-01-13     87
2017-01-14      1

out = df.groupby(df.index.dayofweek)['count'].mean()
print(out)

1    85.0
2    82.0
3    66.5
4    84.5
5    49.0
6    86.0
Name: count, dtype: float64