python - 如何对数据框行进行分组并过滤字符串列表中的所有事件？

Question

情况

有一个df即持有元素，每个元素可以有不同的类型:

df = pd.DataFrame({
    'id':[1,1,2,2,2],
    'type':['a','b','a','b','c']
})

这df应该针对列表进行过滤并仅显示 ids那types 包含所有 list :

['a','b','c']

预期id在这种情况下是:2 导致其所有 types a,b,c 在列表中。

我尝试过的方法

按 id 分组并获取其 types 的列表，但不知道如何与 list 进行比较或应用类似 set(list).issubset(types) 的东西:

df = df.groupby('id')['type'].apply(list)
df.reset_index(name = 'types')

查询list , 但结果都是 types在:

df.query('type in ["a","b","c"]')

很高兴得到提示 - 谢谢

Answer 1

你可以尝试 set.issubset 和 groupby.apply

l = ['a','b','c']
s = df.groupby("id")['type'].apply(lambda x: set(l).issubset(x))
s[s].index

---

Int64Index([2], dtype='int64', name='id')

---

s[s].index[0]
#2