使用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)