python - 如何查找数据帧中的值及其后面的值？

Question

我有一个 pandas 数据框，其中包含河流水位和降雨量，并位于名为“每小时”的数据框中。我希望能够循环遍历每个降雨量值并收集其河流水位值以及下一个河流水位读数。例如:

River Level Rainfall
0.876       0.0
0.877       0.8
0.882       0.0

在这种情况下，如果我正在查找 0.8 毫米降雨量的值，我希望它返回与 0.8 位于同一行的 0.877，以及紧随其后的行中的 0.882。我希望它输出:

0.877
0.882

目前，我有一个循环，可以遍历并定位给定降雨值的所有行，但无法弄清楚如何获取值后的一行。

非常感谢任何帮助。

Answer 1

试试这个

s = df.Rainfall.eq(0.8)
out = df.loc[s | s.shift(), 'River Level']

Out[364]:
1    0.877
2    0.882
Name: River Level, dtype: float64