可以使用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回归
                
下一篇:按两列分组计算统计量的大小