按bar与baz的唯一组合分组提取foo的唯一值列表

嗨，我来帮你搞定这个需求！你想要把DataFrame里的foo值按照bar和baz的唯一配对来分组，每组保留不重复的foo，最后输出成列表的列表对吧？用Pandas几行代码就能轻松实现。

实现代码 #

假设你的DataFrame变量名为df，可以这样写：

import pandas as pd

# 按bar和baz的组合分组，提取每组唯一的foo值并转成列表的列表
foo_list = df.groupby(['bar', 'baz'])['foo'].unique().apply(list).tolist()

代码逻辑拆解 #

• 分组：groupby(['bar', 'baz'])会把整个DataFrame按照bar和baz的所有唯一配对来划分组，比如示例里就会分成(1234, 312)和(0777, 666)两个组
• 去重：['foo'].unique()针对每个组里的foo列，提取出所有不重复的值
• 格式转换：apply(list)把每组的去重结果从数组转成普通列表，最后.tolist()把分组后的结果整体转成你需要的列表嵌套格式

运行这段代码后，得到的foo_list就和你示例里想要的结果完全一致：

[['1234_312_GCD', '1234_312_lcm'], ['0777_666_lcm', '0777_666_GCD']]

如果你的bar或baz是带前导零的字符串（比如示例里的0777）也不用担心，Pandas的groupby会正确识别这些字符串组合，不会把它们当成数值处理。

备注：内容来源于stack exchange，提问作者DeltaIV