如何使用.format在python DataFrame中創建新列

Question

我有一個列表代碼[]，其元素是隨機整數，應該是新DataFrame的列標題。

for i in range (len(codes)):
       df['col{}'.format(codes[i])] 

給我：KeyError：'col1'

有人可以告訴我是否有任何方法可以糾正此問題，或者可以告訴我另一種將這些列表元素轉換為新列的方法嗎？

Answer 1

實例化數據框時，您始終可以設置新的列名：

import pandas as pd

df = pd.DataFrame(columns=["col{}".format(_code) for _code in codes])

Answer 2

或將map與__add__一起__add__ ：

df.columns = map('col'.__add__, codes)

或使用以下format map ：

df.columns = map('col{}'.format, codes)