假设您有一个 pandas 数据框输入和一个列名称列表。获取这些名称表示的列的索引(列号)列表的好方法是什么?
我认为它会是这样的:
def get_col_indices(df, names):
return [df.columns.index(name) for name in names]
Run Code Online (Sandbox Code Playgroud)
def get_col_indices(df, names):
return df.columns.get_indexer(names)
Run Code Online (Sandbox Code Playgroud)