python - 当数据框的列包含多个值时选择数据框的行

Question

我有一个 pandas 数据框，其中“genres”列有多个由 “|” 分隔的值。我在下面放了一张图片。

包含电影详细信息的数据框:

如果我使用 split 函数，它将被转换为不可散列的列表。

现在我只想在"genres" 中包含单词"Action" 时选择数据框的行？我该怎么做？

提前致谢。

Answer 1

这是使用 set 的一种解决方案:

df = pd.DataFrame({'genres': ['A|B|C|D', 'A|B|C', 'B|D']})

res = df[df['genres'].str.split('|').apply(set) >= {'D'}]

print(res)

    genres
0  A|B|C|D
2      B|D

这可以自然地扩展到多种流派:

res = df[df['genres'].str.split('|').apply(set) >= {'A', 'B'}]

print(res)

    genres
0  A|B|C|D
1    A|B|C