パンダでは、リストを問い合わせる方法は？

Question

はのは、私がその値としてリストを持っているデータフレームを持っているとしましょう：

df = pd.DataFrame({'languages': [['en'], ['fr']], 'author': ['Dickens, Charles', 'Austen, Jane']}) 

私は簡単に文字列のためにそれを照会することができます

正しくその基準に一致 dfのサブセットを返し

df[df['author'] == 'Dickens, Charles'] 

。しかし、私はこのような値['en']のようなものですlanguagesとしてリストされているセルの内容を、持っているとき、私はそれを検索するために見えることはできません。

df[df['languages'] == ['en']] 

私が手：

ValueError: Arrays were different lengths: 2 vs 1 

どのようにすることができますがリストであるコンテンツのクエリ？

Answer 1

はlanguages列をループしapplyメソッドを使用して、項目を比較です。これにより、applyの使用が回避されることに注意してください。

# create a numpy array of lists... one list to be exact 
c = np.empty(1, object) 
c[0] = ['en'] 

df[df.languages.values == c] 

      author languages 
0 Dickens, Charles  [en] 

Answer 2

私は通常、isin()フィルタを使用し、引数としてリストを渡します。

df[df.languages.apply(lambda x: x == ['en'])] 
#    author languages 
#0 Dickens, Charles [en] 

Answer 3

私たちは、これを高速に実行するために取得するには、いくつかの策略を使用することができます：あなたが行う可能性がありますどのような

lst = ['A', 'B'] 
df[df['column'].isin(lst)]