我有一个带有Multindex( id 和 datetime )和一个名为X1的列的熊猫数据框。
X1
id datetime
a1ssjdldf 2019 Jul 10 2
2019 Jul 11 22
2019 Jul 12 21
r2dffs 2019 Jul 10 14
2019 Jul 11 13
2019 Jul 12 11
我想创建一个新变量X2,其中相应的值是同一行的X1值和上一行的X1值之差。但是,每次看到新的ID时,都必须从零重新启动相应的值。
例如:
X1 X2
id datetime
a1ssjdldf 2019 Jul 10 2 0
2019 Jul 11 22 20
2019 Jul 12 21 -1
r2dffs 2019 Jul 10 14 0
2019 Jul 11 13 -1
2019 Jul 12 11 -2
答案 0 :(得分:2)
在第一级使用DataFrameGroupBy.diff
,并用Series.fillna
替换缺少的值:
df['X2'] = df.groupby(level=0)['X1'].diff().fillna(0, downcast='int')
print (df)
X1 X2
id datetime
a1ssjdldf 2019 Jul 10 2 0
2019 Jul 11 22 20
2019 Jul 12 21 -1
r2dffs 2019 Jul 10 14 0
2019 Jul 11 13 -1
2019 Jul 12 11 -2