根据 Python 中两列或多列的条件从上一行或下一行中获取值

Question

我正在使用 Pandas Python 来计算行之间的时间增量，当它使用前一行或下一行时，它是基于条件的。

我的桌子是这样的 sampel table

我想用这个条件创建 timedelta 列： 当 STATUS > status_before 时，它​​会从之前的行中获取值 并在 STATUS

我尝试了几种方法来做到这一点，大多数都以这个错误结束：

The truth value of a Series is ambiguous. Use a.empty, a.bool(),a.item(), a.any() or a.all().

这是我所做的一个例子：

if db.STATUS == 0 and db.status_before == 1:
   db.delta = db['REPORTDATE'].shift()

有什么解决办法吗？

Answer 1

• 要获得更高质量的答案，请始终以文本而非图像的形式提供示例数据
• np.where() 允许您在赋值中使用条件逻辑
• 你的逻辑没有指定当条件不满足时使用什么值，所以我使用了 REPORTDATE

import numpy as np

df = pd.DataFrame({"REPORTDATE":pd.date_range("2021-06-18", freq="8h", periods=5), "NAMA":["WPP 571"]*5, 
              "PELABUHAN":(["PP Belawan","di laut"]*3)[1:6], "STATUS":([1,0]*3)[0:5],
             "ASAL":(["PP Belawan","di laut"]*3)[0:5], "IGNORED":[0]*5,
             "INSERTTIME":[pd.to_datetime("2021-06-24 09:26:35")]*5})



(
    df.assign(
        status_after=df["STATUS"].shift(-1), status_before=df["STATUS"].shift()
    ).assign(
        timedelta=lambda d: np.where(
            (d.STATUS == 0) & (d["status_before"] == 1),
            d["REPORTDATE"].shift(),
            d["REPORTDATE"],
        )
    )
)

输出

	REPORTDATE	NAMA	PELABUHAN	STATUS	ASAL	IGNORED	INSERTTIME	status_after	status_before	timedelta
0	2021-06-18 00:00:00	WPP 571	di laut	1	PP Belawan	0	2021-06-24 09:26:35	0	nan	2021-06-18 00:00:00
1	2021-06-18 08:00:00	WPP 571	PP Belawan	0	di laut	0	2021-06-24 09:26:35	1	1	2021-06-18 00:00:00
2	2021-06-18 16:00:00	WPP 571	di laut	1	PP Belawan	0	2021-06-24 09:26:35	0	0	2021-06-18 16:00:00
3	2021-06-19 00:00:00	WPP 571	PP Belawan	0	di laut	0	2021-06-24 09:26:35	1	1	2021-06-18 16:00:00
4	2021-06-19 08:00:00	WPP 571	di laut	1	PP Belawan	0	2021-06-24 09:26:35	nan	0	2021-06-19 08:00:00