PYTHON如何使用numpy.where添加列，其中包括下一行数据帧中的数据？

我是python新手，非常感谢您的帮助

我试图在python中向数据帧添加一列。我正在使用以下命令：

df['capital']=np.where(df['year']!=1960,2,df['GDP'])

除了，当我写GDP时，我真正想要的是1961年的GDP。你知道我该如何包含它吗？

你可以使用

.shift（）

来使用下一行的

GDP

值。默认情况下，将

1

传递给

.shift（）

，但您也可以传递

-1

或

2

等，以便根据需要向上或向下比较多行：

df['capital'] = np.where((df['year'] != 1960), 2, df['GDP'].shift())