2016-08-24 10 views
1

私は数百万の行を持つデータフレームを持っています。元のデータフレームに存在する数十万の要素の対応するインデックスを取得する必要があります。Pythonのデータフレームの要素のリストに対応するインデックスのリストを見つける

私は現在、このコードを使用します。

df[df['processed_col'] == element.index[0] 

全体のデータフレームに'element'の位置を見つけるために。

の代わりに非常に長いループをやって、対応するインデックスのリストを返す入力としてelement1, element2,..., elementNようなリストを取るための方法がある:あなたがisinが必要

df[df['processed_col'] == [element1, element2, ..., elementN].index[0] 
+0

うーんを、あなたはDF [。] 'processed_col' [DF ISIN([要素1、要素2、...、elementN])]インデックス 'を必要とする[0 ] ' – jezrael

+0

申し訳ありませんあなたの質問はあなたが' df [df ['processed_col']。isin(some_list)]。 – EdChum

+0

@jezrael私はOPが値が存在するインデックス値を望んでいると思います、 'index [0]'は最初のインデックス値を返します – EdChum

答えて

0

IIUC最初に発見した場合

df[df['processed_col'].isin([element1, element2,..., elementN])].index 
:インデックスのすべての値をしたい場合

df[df['processed_col'].isin([element1, element2,..., elementN])].index[0] 

または、[0]のみを削除:インデックスの値

使用tolistリストに変換する必要がある場合:。

df[df['processed_col'].isin([element1, element2,..., elementN])].index.tolist() 
関連する問題