python - 为多索引数据帧的第一行分配一个值

Question

我有一个带有多级索引的数据框，看起来有点像这样:

                 value  diffs 
 ticker   date              
   A      22     0.55    NaN
          32    -2.50  -3.05
          76    -0.79  -0.56
   B      59    -1.01  -0.22
          72    -1.24  -0.23
   C      22    -1.29  -0.05
          63     1.65   2.94

如何更改每个级别的第一行以便我可以获得:

                  value  diffs 
 ticker   date              
   A      22     0.55    0
          32    -2.50  -3.05
          76    -0.79  -0.56
   B      59    -1.01    0
          72    -1.24  -0.23
   C      22    -1.29    0
          63     1.65   2.94

Answer 1

您可以使用groupby，例如

df['diffs'] = df.values.groupby(level=0).apply(lambda x: (x-x.shift(1)).fillna(0))