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

yftpprvb 于 2022-10-23 发布在其他

关注(0)|答案(2)|浏览(129)

我想得到Pandas Dataframe 列的数字索引。
只需一列，它就非常简单：

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 ]

更好的主意？

pandas

来源：https://stackoverflow.com/questions/22934609/get-indexes-of-multiple-pandas-columns-by-names

2条答案

按热度按时间

ct2axkht1#

这是有效的，而且还保留了顺序：

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

列出理解是你的朋友。

赞(0）回复(0）举报 2022-10-23

yws3nbqq2#

df=pd.DataFrame(columns=['conditionF', 'conditionB', 'conditionA', 'conditionD', 'conditionC'])
def search():
    search=['conditionA', 'conditionB', 'conditionC']
    c=len(search)
    for i in df.columns.values:
        print(i)
        if i in search:
            c-=1
    return c==0
search()

赞(0）回复(0）举报 2022-10-23

我来回答

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

2条答案

相关问题

热门标签

最新问答