计算数据帧中的平均值和标准偏差

2024-04-19 23:48:36 发布

您现在位置:Python中文网/ 问答频道 /正文

我有以下数据帧:

    COD     CHM     DATE
0   5713    0.0     2020-07-16
1   5713    1.0     2020-08-11
2   5713    2.0     2020-06-20
3   5713    3.0     2020-06-19
4   5713    4.0     2020-06-01
... ... ... ...
2135283 73306036    0.0     2020-09-30
2135284 73306055    12.0    2020-09-30
2135285 73306479    9.0     2020-09-30
2135286 73306656    3.0     2020-09-30
2135287 73306676    1.0     2020-09-30

我想计算整个日期(时间)内每个COD的平均值和标准偏差。 为此,我正在做:

    traf_user_chm_med =traf_user_chm_med.groupby(['COD', 'DATE'])['CHM'].sum().reset_index()
dates = pd.date_range(start=traf_user_chm_med.DATE.min(), end=traf_user_chm_med.DATE.max(), freq='MS', closed='left').sort_values(ascending=False)
clients = traf_user_chm_med['COD'].unique()
idx = pd.MultiIndex.from_product((clients, dates), names=['COD', 'DATE'])
M0 = pd.to_datetime('2020-08')
M1 = M0-pd.DateOffset(month=M0.month-1)
M2 = M0-pd.DateOffset(month=M0.month-2)
M3 = M0-pd.DateOffset(month=M0.month-3)
M4 = M0-pd.DateOffset(month=M0.month-4)
M5 = M0-pd.DateOffset(month=M0.month-5)
def filter_dates(grp):
    grp.set_index('YEAR_MONTH', inplace=True)
    grp=grp[M0:M5].reset_index()
    return grp
traf_user_chm_med = traf_user_chm_med.groupby('COD').apply(filter_dates)

不确定它为什么不工作,它返回一个空数据帧。 在这之后,我将取消堆栈以获得几个月内的活动,并计算每个COD的平均值和标准偏差。 这是一个漫长的过程,不确定是否有更快的方法让我得到想要的值。 尽管如此,如果有人能帮我让这台机器运转起来,那真是太棒了


Tags: 数据dateindexmedcodpddatesuser
3条回答
df['mean'] = df.groupby('DATE')['COD'].transform('mean')

您可以使用transform来广播您的平均值和std

...
df['mean'] = df.groupby('DATE')['COD'].transform('mean')
df['std'] = df.groupby('DATE')['COD'].transform('std')

如果我理解正确,您只需要:

df.groupby("COD")["CHM"].agg("std")

作为一个一般原则,几乎总是有一种“pythonic”的方式来做这些事情,它的行数更少,也更容易理解

相关问题 更多 >