使用列表作为熊猫数据框中的值

Question

我有一些数据要插入到数据框中。数据为columns= ['Title', 'Category']。对于每个标题，我都有一个或多个类别，我决定将类别作为列表插入。所以我的 df 看起来像这样：

In [39]: title_cat_df
Out[39]: 
    Title      Category
0  Title1  [Cat1, Cat2]
1  Title3        [Cat5]
2  Title2  [Cat3, Cat4]
...
...
...

但是我不知道这是否是 pythonic/pandaionic(?!) 方法，因为我偶然发现了一些问题，例如使用 isin 查找特定类别：

In [41]: test_df['Category'].isin(cat_list)
Out[41]: TypeError: unhashable type: 'list'

在这种情况下，表示类别的更好方法是什么，并希望能够在特定类别中查找标题？

Answer 1

将列转换为sets 并使用& 与转换为set 的列表的交集：

cat_list = ['Cat1','Cat2', 'Cat4']
print (test_df['Category'].apply(set) & set(cat_list))
0     True
1    False
2     True
Name: Category, dtype: bool

boolean indexing的最后过滤器：

test_df = test_df[test_df['Category'].apply(set) & set(cat_list)]
print (test_df)
    Title      Category
0  Title1  [Cat1, Cat2]
2  Title2  [Cat3, Cat4]