使用循環重命名數據框中的列

Question

假設我有一個數據框（df），如下所示

它有 3 列都命名相同（L，L，L）。 我想根據列位置將列重命名為 L1、L2 和 L3。 我希望 optput 如下所示。

我可以使用代碼 df.columns = ['Input', 'L1', 'L2', 'L3']

但是，我相信必須有一種方法可以循環遍歷所有列（或選擇的列），以便可以為任何數據框自動執行此操作。

非常感謝您的回復。

Answer 1

你用

df.columns = [f'{col}{i}' if col == 'L' else col for i, col in enumerate(df.columns)]