要按照一个列进行分组后的移动平均值,可以使用pandas库来处理。下面是一个示例代码:
import pandas as pd
# 创建一个示例DataFrame
data = {'group': ['A', 'A', 'A', 'B', 'B', 'B', 'B'],
'value': [1, 2, 3, 4, 5, 6, 7]}
df = pd.DataFrame(data)
# 按照'group'列进行分组,并计算移动平均值
df['moving_average'] = df.groupby('group')['value'].rolling(window=2).mean().reset_index(0, drop=True)
print(df)
上述代码首先创建了一个示例DataFrame,其中包含了一个'group'列和一个'value'列。然后,使用groupby
方法按照'group'列进行分组,然后使用rolling
方法计算每个分组中'value'列的移动平均值。在这个示例中,我们使用了窗口大小为2的移动平均。最后,我们将计算得到的移动平均值赋值给新的列'moving_average'。
运行代码后,输出的DataFrame将包含原始数据以及每个分组的移动平均值。