使用python pandas在特定列中使用最大值在數據框中添加新的“x”列數

Question

我有一個熊貓數據框，如下所示：

從這樣的 DataFrame 我想生成以下輸出：

為了獲得所需的數據幀，我使用了下面的代碼。

first_list_len_max = df1['first_list_len'].max()

它給了我需要在數據框中添加的最大列數，然后我手動添加了 4 列，如下所示。

df1[['1_','2_','3_','4_']] = pd.DataFrame(df1.first_list.values.tolist(), index= df1.index)

在這里，我需要一些自動方法來使用first_list_len_max添加列first_list_len_max 。

備注： first_list_len_max值可能每次都改變。

提前致謝！！

Answer 1

對f-string使用列表理解：

first_list_len_max = df1['first_list_len'].max()
L = [f'{i}_' for i in range(1, first_list_len_max + 1)]
df1[L] = pd.DataFrame(df1.first_list.values.tolist(), index= df1.index)