如何根据 Jupyter 中的现有列在数据框中创建新列？

Question

假设我有一个像这样的数据框：

    A   B
0   1   3
1   2   4
2   3   5
3   4   6

我想要一个等于 A[i-1] 和一个等于 B[i-1] 的新列，所以：

    A   B   A1   B1
0   1   3   Nan  Nan
1   2   4   1    3
2   3   5   2    4
3   4   6   3    5

我该怎么做？

Answer 1

IIUC，

一个想法是将列传递给列表并循环，同时将您的 col 分配给 shift()

cols = df.columns.tolist()
for col in cols:
    df[f'{col}1'] = df[col].shift(1)
print(df)

    A   B   A1  B1
0   1   3   NaN NaN
1   2   4   1.0 3.0
2   3   5   2.0 4.0
3   4   6   3.0 5.0