Python에서 팬더 데이터 프레임에 대한 행 추출하기

Question

파이썬의 팬더 데이터 프레임에 간단한 쿼리 로그를 가져 왔으며 (이미지 참조), 주어진 쿼리를 포함하는 모든 행을 가장 효율적으로 추출하는 방법을 알고 싶습니다 '키워드'열에 포함 된 키워드입니다.

데이터 프레임을 반복 할 수 있지만 배열/마스크를 사용하면 더 빠른 방법 일 수 있습니다.

도움을 주시면 감사하겠습니다.

Answer 1

당신은 예를 들어, str.contains를 사용할 수 있습니다

In [1]: df = pd.DataFrame([['abc', 1], ['cde', 2], ['efg', 3]]) 

In [2]: df 
Out[2]: 
    0 1 
0 abc 1 
1 cde 2 
2 efg 3 

In [3]: df[0].str.contains('c') 
Out[3]: 
0  True 
1  True 
2 False 
Name: 0, dtype: bool 

In [4]: df[df[0].str.contains('c')] 
Out[4]: 
    0 1 
0 abc 1 
1 cde 2 

Answer 2

[3]: df = DataFrame(dict(A = ['foo','bar','bah','bad','bar'],B = range(5))) 

In [4]: df 
Out[4]: 
    A B 
0 foo 0 
1 bar 1 
2 bah 2 
3 bad 3 
4 bar 4 

In [5]: select = Series(['bar','bah']) 

In [6]: df[df.A.isin(select)] 
Out[6]: 
    A B 
1 bar 1 
2 bah 2 
4 bar 4