web-dev-qa-db-ja.com

Pythonパンダの列名から列インデックスを取得する

Rでは、列の名前に基づいて列インデックスを取得する必要があるとき

idx <- which(names(my_data)==my_colum_name)

パンダデータフレームで同じことをする方法はありますか？

pythonpandasdataframeindexing
140
ak3nat0n

.get_loc()を使うことができます。

In [45]: df = DataFrame({"pear": [1,2,3], "Apple": [2,3,4], "orange": [3,4,5]})

In [46]: df.columns
Out[46]: Index([Apple, orange, pear], dtype=object)

In [47]: df.columns.get_loc("pear")
Out[47]: 2

正直なところ、私はこれを自分自身で必要とすることはあまりありません。私はあなたがインデックス番号が欲しいケースを確かに見ることができるけれども、通常名前によるアクセスは私がそれが望むもの（df["pear"]、df[["Apple", "orange"]]、あるいはdf.columns.isin(["orange", "pear"])）を行います。

229
DSM

これはリスト内包表記による解決策です。 colsは、インデックスを取得する列のリストです。

[df.columns.get_loc(c) for c in cols if c in df]
14
snovik

DSMのソリューションはうまくいきますが、whichと直接同等のものが欲しい場合は(df.columns == name).nonzero()を実行できます。

13
Wes McKinney

複数の列の一致を探したい場合は、 searchsorted method を使用したベクトル化解法を使用できます。したがって、データフレームとしてdf、検索対象の列名としてquery_colsを使用すると、実装は次のようになります。

def column_index(df, query_cols):
    cols = df.columns.values
    sidx = np.argsort(cols)
    return sidx[np.searchsorted(cols,query_cols,sorter=sidx)]

サンプル実行 - 

In [162]: df
Out[162]: 
   Apple  banana  pear  orange  Peach
0      8       3     4       4      2
1      4       4     3       0      1
2      1       2     6       8      1

In [163]: column_index(df, ['Peach', 'banana', 'Apple'])
Out[163]: array([4, 1, 0])
8
Divakar

列の位置から列名を取得したい場合（逆にOP質問への逆の方法）、以下を使用できます。

>>> df.columns.get_values()[location]

@DSMの使用例

>>> df = DataFrame({"pear": [1,2,3], "Apple": [2,3,4], "orange": [3,4,5]})

>>> df.columns

Index(['Apple', 'orange', 'pear'], dtype='object')

>>> df.columns.get_values()[1]

'orange'

別の方法：

df.iloc[:,1].name
6
salhin