Python - l oop имена столбцов

Question

Я не могу найти решение своей проблемы. У меня есть фрейм данных, например

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, в которых я хотел бы определить ячейки на основе функции xi, где x - это значение ячейки, а i - это шаг итерации (1, .., 5) - было бы достаточно иметь желаемый результат для столбца B. B_1 должно быть [6,7,8,3,1,2] et c .. Можете ли вы дать мне какие-то намеки?

Спасибо

Answer 1

В случае наличия пяти итераций и их известности заранее:

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