python - 如何在 DataFrame 中创建显示最后记录峰值的列？

Question

我正在尝试创建一个新列来列出最后记录的峰值，直到下一个峰值出现。例如，假设这是我现有的 DataFrame:

index values
0      10
1      20
2      15
3      17
4      15
5      22
6      20

我想得到这样的东西:

index values last_recorded_peak
0      10        10
1      20        20
2      15        20
3      17        17
4      15        17
5      22        22
6      20        22

到目前为止，我已尝试使用 np.maximum.accumulate，它“累积”最大值但不完全是“峰值”(某些峰值可能低于最大值)。

我也尝试过使用 scipy.signal.find_peaks 返回我的峰值所在的索引数组(在示例中，索引 1、3、5)，这不是我要找的。

我对编码比较陌生，非常感谢任何指点!

Answer 1

你在正确的轨道上，scipy.signal.find_peaks是我要走的路，你只需要根据结果做一点工作:

from scipy import signal
peaks = signal.find_peaks(df['values'])[0]

df['last_recorded_peak'] = (df.assign(last_recorded_peak=float('nan'))
                              .last_recorded_peak
                              .combine_first(df.loc[peaks,'values'])
                              .ffill()
                              .combine_first(df['values']))

---

print(df)

     index  values  last_recorded_peak
0      0      10                10.0
1      1      20                20.0
2      2      15                20.0
3      3      17                17.0
4      4      15                17.0
5      5      22                22.0
6      6      20                22.0