按组计算行中值之间的差异

Question

我有这样的df：

我需要为每家公司计算X1, X2, X3, X4 and X5 之间的差异，但我只知道如何计算整列之间的差异。

df['dX1'] = df['X1'].shift(-1) - df['X1']
df['dX2'] = df['X2'].shift(-1) - df['X2']
df['dX3'] = df['X3'].shift(-1) - df['X3']
...

这是一种不好的方法，因为它在74 上用X1 在73 上减去X1（这是两个不同的公司，这没有任何意义）。

我的问题是，如何计算每个时期和每个公司的行值之间的差异。例如。

（我可以一个一个加载csv并计算每个公司的差异，然后将所有合并到一个列表中，但是因为我有700个csv文件，所以需要2天）。

Answer 1

我觉得你需要DataFrameGroupBy.diff:

cols = ['X1','X2','X3','X4','X5']
df1 = df.groupby('company')[cols].diff(-1)