如何根據輸入數字添加多個新的 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)})