python - 如何获得 Pandas 第一行和当前行之间的区别

Question

我在数据框中有 2 列(日期和销售价格)。我的预期输出是这样的。我想在 dataframe 中再添加一个名为 profit 的列，需要通过当前 sell_price - first sell price (stars one) 计算

        date            sell_price  profit(needs to be added)
    0   2018-10-26       **21.20**    NaN
    1   2018-10-29       15.15      -6.05
    2   2018-10-30       15.65      -5.55
    3   2018-10-31        0.15     -21.05
    4   2018-11-01        5.20     -16.00

我知道 pandas 中的差异会导致连续行之间存在差异。我们如何在 pandas 上使用 diff 或任何其他函数实现预期的 o/p？

Answer 1

对于像DatetimeIndex 这样的一般Index 使用iloc与 iat , 但它只适用于职位，所以有必要 get_loc :

pos = df.columns.get_loc('sell_price')
df['profit'] =  df.iloc[1:, pos] - df.iat[0, pos]

如果默认 RangeIndex 使用 loc与 at :

df['profit'] =  df.loc[1:, 'sell_price'] - df.at[0, 'sell_price']

---

print (df)
         date  sell_price  profit
0  2018-10-26       21.20     NaN
1  2018-10-29       15.15   -6.05
2  2018-10-30       15.65   -5.55
3  2018-10-31        0.15  -21.05
4  2018-11-01        5.20  -16.00