python - Pandas 在日期范围内的按和均值

我想使用groupby和mean计算每个ID的平均值。但是，我只需要Date在年份2016-01-01和2017-12-31之间的行。

d = {'ID': ['STCK123', 'STCK123', 'STCK123'], 'Amount': [250, 400, 350],
     'Date': ['2016-01-20', '2017-09-25', '2018-05-15']}



data = pd.DataFrame(data=d)
data = data[['ID', 'Amount', 'Date']]

data['Date'] = pd.to_datetime(data['Date'])

这给出以下df：

    ID  Amount    Date
STCK123 250 2016-01-20
STCK123 400 2017-09-25
STCK123 350 2018-05-15

当我使用时：

data.groupby(['ID'])['Amount'].agg('mean')

它考虑了所有行，得出平均值333.3。如何排除Date为2018年的行（平均值为(250+400)/2=325）？

最佳答案

您需要使用query进行预过滤的步骤：

df.query('Date.dt.year != 2018').groupby('ID').mean()

         Amount
ID
STCK123     325

eval，query和相关参数的更多用法可以在我的文章中找到：Dynamic Expression Evaluation in pandas using pd.eval()

有关在调用groupby之前删除行的更多方法，请参见here。



您也可以mask那些行，而不必删除它们。 NaN从GroupBy聚合中排除。

df.mask(df.Date.dt.year == 2018).groupby('ID').mean()

         Amount
ID
STCK123   325.0

关于python - Pandas 在日期范围内的按和均值，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/54280746/