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

Question

data['Predicted Order'] = predicted_orderprint(data)data.to_csv('smallys_ARMA.csv', encoding='utf-8', index=False)Pandas 数据帧data在数据帧中获得一个新列Predicted Order，其值被分配给列表predicted_order。在这里，我不想对新生成的列名进行硬编码，现在它是Predicted Order. 我希望每个新生成的列都像这样动态地命名Week_ii = 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)]改用