python - Pandas key 错误 : 'occurred at index 0'

Question

假设我有一个 Pandas 数据框 df:

start_time   Event
0            0
1            0
2            0
3            0
4            0
5            0
6            0
7            0
8            0
9            0

当相应的 start_time 位于两个值之间时，我想将 Event 列的值设置为 -1，因此我定义了以下函数:

def test(time):
    if (time['start_time'] >= 5) and (time['start_time'] <= 8):
        return -1
    else:
        return time

要将其应用到事件列，我执行以下操作:

df[['Event']] = df[['Event']].apply(test,axis=1)

这会产生此错误:KeyError: ('start_time', '发生在索引 0')

为什么会发生这种情况？应该是一个简单的修复。

Answer 1

简单地做:

df['Event'] = df.apply(test, axis=1)['Event']