时间戳

首页 > 解决方案 > 在 Pandas 中进行分组和插值

python - 在 Pandas 中进行分组和插值

问题描述

我有包含周数、帐户 ID 和几个使用列的数据。我想a）按帐户ID分组，b）将每周数据重新采样为每日数据，c）平均插入每日数据（每周除以7），然后将它们全部重新组合在一起。我已经把大部分内容都记下来了，但是 Pandasgroupby让我有点困惑。它也很慢，这让我认为这可能不是最佳解决方案。

数据如下所示：

    Account Id  year week         views stats foo_col 
31133   213     2017-03-05          4.0     2.0     11.0
10085   456     2017-03-12          1.0     6.0     3.0
49551   789     2017-03-26          1.0     6.0     27.0

这是我的代码：

def interpolator(mini_df):
    mini_df = mini_df[cols_to_interpolate].set_index('year week')
    return mini_df.resample('D').ffill().interpolate() / 7

example = list(grp)[0][1]
interpolator(example) # This works perfectly

df.groupby('Account Id').agg(interpolator)                # doesn't work
df.groupby('Account Id').transform(interpolator)          # doesn't work

for name,group in grp:
    group = group[cols_to_interpolate].set_index('year week')
    group = group.resample('D').ffill().interpolate() / 7 # doesn't work

for acc_id in df['Account Id'].unique():
    mask = df.loc[df['Account Id'] == acc_id]
    print(df[mask])                                     # doesn't work

标签： pythonpandasinterpolationpandas-groupby

解决方案

我希望你的函数应该与groupby像这样的对象链接：

df = (df.set_index('year week')
        .groupby('Account Id')[cols_to_interpolate]
        .resample('D')
        .ffill()
        .interpolate() / 7)

来自评论的解决方案是不同的 -interpolate适用于每个组：

df.groupby('Account Id').apply(interpolator)

推荐阅读