python - pandas diff() 为一阶差分给出 0 值，我想要实际值

Question

我有 df:

Hour  Energy Wh  
1        4          
2        6           
3        9
4        15

我想添加一个显示每小时差异的列。我正在使用这个:

df['Energy Wh/h'] = df['Energy Wh'].diff().fillna(0)

df1:

Hour  Energy Wh  Energy Wh/h
1        4          0
2        6          2 
3        9          3
4        15         6

但是，第 1 小时的值在 Energy Wh/h 列中显示为 0，而我希望它显示为 4，如下所示:

Hour  Energy Wh  Energy Wh/h
1        4          4
2        6          2 
3        9          3
4        15         6

我试过使用 np.where:

df['Energy Wh/h']  = np.where(df['Hour'] == 1,df['Energy Wh'].diff().fillna(df['Energy Wh']),df['Energy Wh'].diff().fillna(0))

但我在第 1 小时行 (df1) 中仍然得到 0 值，没有错误。如何获取要填充的第 1 小时的“Energy Wh”值，而不是 0？

Answer 1

您可以只对原始列进行 fillna()，而无需使用 np.where:

>>> df['Energy Wh/h'] = df['Energy Wh'].diff().fillna(df['Energy Wh'])
>>> df
      Energy Wh  Energy Wh/h
Hour
   1          4          4.0
   2          6          2.0
   3          9          3.0
   4         15          6.0