Python + Pandas + Dataframe：为Pandas Dataframe列生成动态列名

Question

data['Predicted Order'] = predicted_order
print(data)
data.to_csv('smallys_ARMA.csv', encoding='utf-8', index=False)

Pandas数据帧data在数据帧中获得新列Predicted Order ，其值分配给列表predicted_order 。

在这里，我不想对新生成的列名进行硬编码，现在是Predicted Order 。 我希望每个新生成的列都像Week_i这样动态获取名称，例如i = 0,1,2等。

编辑：此代码将来将运行未知次数，并且每次它将向数据框添加新列。

Answer 1

也许是这样的：

numbers = range(52)
for i in numbers:
    if not "weeks_{i}" in df.columns:
        df["weeks_{i}"] = predicted_order
        break

Answer 2

您可以为此使用字符串格式。

weeks = range(42)

for i in weeks:
     data[f'Week_{i}'] = predicted_order

如果您使用的是Python <3.6，请改用data['Week_{}'.format(i)]