如何在Pandas中将查询结果行作为新列追加至原DataFrame

需求说明 #

遍历输入DataFrame的每一行，对每行执行查询后得到5行结果，需将这些结果分别作为duplicate_materials_0至duplicate_materials_4列，追加到原DataFrame的对应行后。

查询结果示例 #

duplicate_materials
0  [BIAS CUT FABRIC 50 X 1600 MM, RBN20 (000000000001065341)]
1  [BIAS CUT FABRIC 40 X 1600 MM, RBN20 (000000000001061715)]
2  [BIAS CUT  FABRIC 40 X 1100 MM, RBN20 (000000000001059946)]
3  [BIAS CUT FABRIC 40 X 1200 MM, RBN20 (000000000001059947)]
4  [BIAS CUT  FABRIC 50 X 1300 MM (000000000001005101)]

目标DataFrame示例 #

material_id material_description    duplicate_materials_0   duplicate_materials_1   duplicate_materials_2   duplicate_materials_3   duplicate_materials_4
1   000000000001065341  FABRIC CUT BIAS, RBN20 50 X 1600 MM

解决代码 #

for i, row in results.iterrows():
    df_test[f'duplicate_materials_{i}'] = results['duplicate_materials'][i]

内容的提问来源于stack exchange，提问作者Tossibob