按名称获取多个 pandas 列的索引

Question

我想获取一组 pandas 数据框列的数字索引。

一栏很简单：

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

但是有多个元素？ 我有一些有用的东西，但是很冗长，很冗长：

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 ]

更好的想法？

Answer 1

这行得通，而且还保持秩序：

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

列表推导是你的朋友。