按名称获取多个 pandas 列的索引 [英] Get indexes of multiple pandas columns by names

问题描述

我想获取选择的pandas数据框列的数字索引。

I want to get the numerical indexes of a selection of pandas dataframe columns.

使用一列非常简单：

nonzero(df.columns.values == 'conditionA')

但是有多个元素？
我有一些可行的方法，但是又冗长又笨拙：

but with multiple elements? I have something that works but is verbose and hugly:

df = pd.DataFrame(columns=['conditionF', 'conditionB', 'conditionA', 'conditionD', 'conditionC'])

cols_to_find = ['conditionA', 'conditionB', 'conditionC']
[i for i in range(len(df.columns.values)) if df.columns.tolist()[i] in cols_to_find ]

更好的主意吗？

推荐答案

这可行，并且还保留了顺序：

This works, and also preserves order:

[df.columns.get_loc(col) for col in cols_to_find]
[2, 1, 4]

列表理解是您的朋友。

这篇关于按名称获取多个 pandas 列的索引的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！