如何基于字符串值列表对熊猫数据框进行分类？

Question

我有一个超过10万行的dF和几列宽 - 没有什么疯狂的。我试图根据大约4000个字符串的列表对行进行子集划分，但我正在努力弄清楚如何这样做。有没有一种方法来使用类似的子集。

东风看起来像这样

dog_name count 
=================== 
Jenny  2 
Fido   4 
Joey   7 
Yeller  2 

和字符串列表中包含的变量dog_name_list=['Fido', 'Yeller']

我试着沿 df[df['dog_name'].isin(dog_name_list)线的东西，但我得到一个有趣的错误： unhashable type: 'list'

我检查了similar question，docs和this的子集数据通过查看列表中是否存在某个值，但是这让我无所适从，而且我对我所缺少的东西有些困惑。非常感谢别人的建议！

Answer 1

我相信你的狗名列中有一个列表。

这工作得很好：

>>> df[df['dog_name'].isin(['Fido', 'Yeller'])] 
    dog_name count 
1  Fido  4 
3 Yeller  2 

但是，如果你添加一个列表：

df.ix[4] = (['a'], 2) 
>>> df 
    dog_name count 
0 Jenny  2 
1  Fido  4 
2  Joey  7 
3 Yeller  2 
4  [a]  2 

>>> df[df['dog_name'].isin(['Fido', 'Yeller'])] 
--------------------------------------------------------------------------- 
TypeError         Traceback (most recent call last) 
<ipython-input-20-1b68dd948f39> in <module>() 
----> 1 df[df['dog_name'].isin(['Fido', 'Yeller'])] 
... 
pandas/lib.pyx in pandas.lib.ismember (pandas/lib.c:5014)() 

TypeError: unhashable type: 'list' 

为了找到这些坏狗：

>>> df[[isinstance(dog, list) for dog in df.dog_name]] 
    dog_name count 
4  [a]  2 

若要查找的所有数据类型列：

>>> set((type(dog) for dog in df.dog_name)) 
{list, str}