Pandas 数据框每天重新采样和计数事件

Question

我有一个带有时间索引的数据框。我可以对数据重新采样以获取（例如）每天的平均值，但是我也想获取每天的计数。这是一个示例：

import datetime
import pandas as pd
import numpy as np

dates = pd.date_range(datetime.datetime(2012, 4, 5, 11, 
0),datetime.datetime(2012, 4, 7, 7, 0),freq='5H')

var1 = np.random.sample(dates.size) * 10.0
var2 = np.random.sample(dates.size) * 10.0
df = pd.DataFrame(data={'var1': var1, 'var2': var2}, index=dates)

df1=df.resample('D').mean()

我还想获得第 3 列 'count' 每天计数：

count
3
5
7

非常感谢！

Answer 1

使用 Resampler.agg，然后在列中展平 MultiIndex：

df1 = df.resample('D').agg({'var1': 'mean','var2': ['mean', 'size']}) 
df1.columns = df1.columns.map('_'.join)
df1 = df1.rename(columns={'var2_size':'count'})
print (df1)
            var1_mean  var2_mean  count
2012-04-05   3.992166   4.968410      3
2012-04-06   6.843105   6.193568      5
2012-04-07   4.568436   3.135089      1

Grouper 的替代解决方案：

df1 = df.groupby(pd.Grouper(freq='D')).agg({'var1': 'mean','var2': ['mean', 'size']}) 
df1.columns = df1.columns.map('_'.join)
df1 = df1.rename(columns={'var2_size':'count'})
print (df1)
            var1_mean  var2_mean  count
2012-04-05   3.992166   4.968410      3
2012-04-06   6.843105   6.193568      5
2012-04-07   4.568436   3.135089      1

编辑：

r = df.resample('D')
df1 = r.mean().add_suffix('_mean').join(r.size().rename('count'))
print (df1)
            var1_mean  var2_mean  count
2012-04-05   7.840487   6.885030      3
2012-04-06   4.762477   5.091455      5
2012-04-07   2.702414   6.046200      1

Pandas 数据框每天重新采样和计数事件

Pandas dataframe resample and count events per day

python

datetime

time-series

resampling

pandas