Python Pandas：从给定名称列表中获取列索引

假设您有一个数据帧输入和一个列名列表。为这些名称所代表的列获取索引列表（列编号）的好方法是什么

我想大概是这样的：

def get_col_索引（df，名称）：
返回[df.columns.index（name）for name in names]

使用：

---

如果您不需要它们处于相同的顺序，您可以使用

---

我总是喜欢遇到另一个我以前从未使用过的

np

函数

def get_col_indices(df, names):
    return df.columns.get_indexer(names)

import numpy as np
np.flatnonzero(df.columns.isin(names))