如何在数据帧上循环，根据列的每个唯一值将某些字段写入文件，并根据该列命名文件

Question

我有一个用于多个ID的组合表。

颜色尺寸Wht><2>Wht[/tr>3

ID	类型	数量
1	A	S	7
1		B	Ylw	L	9
2	A	Grn	S
2	C	M	7
3	B	Blk	M

您可以使用：

# Enhanced by @piRSquared
for name, subdf in df.groupby('ID'):
data = subdf['Type'] + subdf['Color']
data.to_csv(f'{name}.csv', index=False, header=None)
# Old answer
# for name, subdf in df.groupby('ID'):
#     with open(f'{name}.txt', 'w') as fp:
#         data = subdf['Type'] + subdf['Color']
#         data.to_csv(fp, index=False, header=None)

输出：

>>> %cat 1.txt
AWht
BYlw
>>> %cat 2.txt
AGrn
CWht
>>> %cat 3.txt
BBlk