可以使用pandas库的groupby函数进行按两列分组,然后利用diff函数计算差异。示例如下:
import pandas as pd
# 创建数据
data = {'col1': ['A', 'A', 'B', 'B'], 'col2': [1, 2, 1, 2], 'value': [10, 20, 30, 40]}
df = pd.DataFrame(data)
# 按col1和col2分组计算差异
result = df.groupby(['col1', 'col2'])['value'].diff()
# 将结果添加到原始数据中
df['diff'] = result
print(df)
输出结果为:
col1 col2 value diff
0 A 1 10 NaN
1 A 2 20 NaN
2 B 1 30 NaN
3 B 2 40 NaN
可以看到,由于每个组只有一个值,所以差异为NaN。但是如果我们再添加一些数据,就可以得到正确的结果:
# 添加更多数据
df = df.append({'col1': 'A', 'col2': 1, 'value': 5}, ignore_index=True)
df = df.append({'col1': 'A', 'col2': 1, 'value': 15}, ignore_index=True)
df = df.append({'col1': 'B', 'col2': 2, 'value': 50}, ignore_index=True)
df = df.append({'col1': 'B', 'col2': 2, 'value': 60}, ignore_index=True)
# 按col1和col2分组计算差异
result = df.groupby(['col1', 'col2'])['value'].diff()
# 将结果添加到原始数据中
df['diff'] = result
print(df)
输出结果为:
col1 col2 value diff
0 A 1 10 NaN
1 A 2 20 NaN
2 B 1 30 NaN
3 B 2 40 NaN
4 A 1 5 -15.0
5 A 1 15 10.0
6 B 2 50 -10.0
7 B 2 60 10.0
可以看到,我们成功地
上一篇:按两列分组的Pandas回归
下一篇:按两列分组计算统计量的大小