Python - 循环列名

Question

我找不到我的问题的解决方案。 我有 dataframe 之类的

import pandas as pd
df = pd.DataFrame({'A':[4,5,4,5,5,4],
                   'B':[7,8,9,4,2,3],
                   'C':[1,3,5,7,1,0]})

我想创建 5 个新列 B_1、B_2、...、B_5，我想在其中定义基于 function xi 的单元格，其中 x 是单元格值，i 是迭代步骤 (1,..,5) - 它足以获得 B 列所需的结果。B_1 应该是 [6,7,8,3,1,2] 等。你能给我一些提示吗？

谢谢

Answer 1

在进行五次迭代并且预先知道它们的情况下：

for col in df.columns:
    for i in range(1, 6):
        df[f'{col}_{i}'] = df[col] - i