在熊猫中使用移位功能数据框功能

Question

我想创建一个比较熊猫数据框中不同行的函数。
我当前的功能如下：

def f(row):
    if row['A'].shift(1) == row['B']:
        val = 0        
    else:
        val = 1
    return val

我收到以下错误消息：

AttributeError: ("'numpy.float64' object has no attribute 'shift'", 'occurred at index 2006-02-28 00:00:00')

我知道该问题发生在数据帧的第一行，因为没有可以移动的行。有人知道如何重写功能吗？任何帮助表示赞赏。

Answer 1

使用pandas，您可以。

(~(df.A.shift()==df.B)).astype(int) # since default of shift is 1