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

adr*_*rin 5 python pandas

假设您有一个 pandas 数据框输入和一个列名称列表。获取这些名称表示的列的索引(列号)列表的好方法是什么?

我认为它会是这样的:

def get_col_indices(df, names):
    return [df.columns.index(name) for name in names]
Run Code Online (Sandbox Code Playgroud)

jez*_*ael 5

使用Index.get_indexer

def get_col_indices(df, names):
    return df.columns.get_indexer(names)
Run Code Online (Sandbox Code Playgroud)