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

Question

我对 python 很陌生，非常感谢您的帮助。我正在尝试在 python 中向我的数据框添加一列。我正在使用以下内容：df['capital']=np.where(df['year']!=1960,2,df['GDP'])不过，当我写 GDP 时，我真正想要的是 1961 年的 GDP。知道如何将其包括在内吗？

Answer 1

您可以使用.shift()下一GDP行中的值。默认情况下 a1被传递给.shift()，但您也可以传递-1或2等来向上或向下比较任意数量的行：

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