2017-03-21 16 views
0

DataFrameサブセットを抽出しようとしていますが、最初の列が特定のリストにあるすべての行が必要です。 例えば、私が持っている:値がリストにある場合、pandasサブセットを抽出します

 t gid 
0 2010.0 1 
1 2020.0 4 
2 2040.0 6 
3 2060.0 6 
4 2090.0 12 
5 2110.0 18 
6 2140.0 1000 

と私はGIDがq_listにあるすべての行のデータフレームをしたい== [1,6,18]

+3

が '' DF [df.gid.isin(q_list)]を使用し、http://pandas.pydata.org/pandas-docs/stable/generated/pandas([ 'isin']をチェックします。 Series.isin.html)と['boolean indexing'](http://pandas.pydata.org/pandas-docs/stable/indexing.html#boolean-indexing) – jezrael

+1

@jezrael素晴らしい!私は45秒から3.68にあなたの2番目の答えを持っています! – Oren

+0

スーパー、それは本当に良いです – jezrael

答えて

0

あなたがインデックスに列を変換し、スマートに使用することができますインデックス:

df.set_index('gid').ix[q_list] 
関連する問題