python pandasリストを含むフィルタリング

Question

私は現在、Python 2.7でPandasを使用しています。私のデータフレームは、次のようになります。

>>> df 
     0 
1 [1, 2] 
2 [2, 3] 
3 [4, 5] 

それは列1の値で行をフィルタすることは可能ですか？たとえば、フィルタ値が2の場合、フィルタは最初の2行を含むデータフレームを返します。

私はすでにいくつかの方法を試しています。私が考えることができる最も良いことは、値が存在する行のインデックスを返すリストの理解を行うことです。次に、インデックスのリストを使ってデータフレームをフィルタリングすることができました。しかし、異なる値で複数回フィルタリングしたい場合、これは非常に遅いでしょう。理想的には、私は、プロセスをスピードアップするために、パンダの機能でビルドを使用するものが好きです。

Answer 1

あなたはboolean indexingを使用することができます。

import pandas as pd 

df = pd.DataFrame({'0':[[1, 2],[2, 3], [4, 5]]}) 
print (df) 
     0 
0 [1, 2] 
1 [2, 3] 
2 [4, 5] 

print (df['0'].apply(lambda x: 2 in x)) 
0  True 
1  True 
2 False 
Name: 0, dtype: bool 

print (df[df['0'].apply(lambda x: 2 in x)]) 
     0 
0 [1, 2] 
1 [2, 3] 

Answer 2

ます。また、リスト内包してブールインデックスを使用することができます。

>>> df[[2 in row for row in df['0']]] 
     0 
0 [1, 2] 
1 [2, 3]