Панды: получить индексы столбцов из списка имен

Question

Допустим, у вас есть входные данные от pandas и список имен столбцов. Как можно получить список индексов (номеров столбцов) для столбцов, представленных этими именами?

Я предполагаю, что это будет что-то вроде:

def get_col_indices(df, names):
    return [df.columns.index(name) for name in names]

Answer 1

Если вам не нужно, чтобы они были в одном и том же порядке, вы можете использовать pd.Index.isin с np.flatnonzero

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

Answer 2

Использование Index.get_indexer:

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