如何根据输入数字添加多个新的 dataframe 列？

Question

我需要一步创建多个新的 dataframe 列。 我的方法是询问用户“需要多少列？” 并且根据用户输入，应该在一个 go 上创建很多新列。

以下是我的代码形式的要求：

df1.insert(len(df1.columns),'Syndata_1997', np.nan)
df1.insert(len(df1.columns),'Syndata_1998', np.nan)
df1.insert(len(df1.columns),'Syndata_1999', np.nan)
df1.insert(len(df1.columns),'Syndata_2000', np.nan)

所以要详细说明，而不是使用插入 function 来一一创建一列，我希望在用户输入 4 时创建 4 个命名和值为 NaN 的列，如下所示：

prompt=int(input('Enter the columns to be created:'))

然后，根据提示值，应创建新列。

请让我知道这是否可以完成或任何其他自动创建列的方法。

Answer 1

您可以使用分配命令 -

num_cols = 10

df.assign(**{f'Syndata_{idx}': np.nan for idx in range(num_cols)})