从熊猫索引中检索列的名称

发布于 2021-01-29 15:14:28

我有一个pandas数据框和该数据框的值的numpy数组。我具有特定列的索引,并且我已经具有重要值的行索引。现在,我需要从数据框中获取该特定值的列名称。

在搜索了文档之后,我发现我可以做相反的事情,但不能做到。

关注者
0
被浏览
48
1 个回答
  • 面试哥
    面试哥 2021-01-29
    为面试而生,有面试问题,就找面试哥。

    我认为您需要按位置排列索引列名称(python从开始计数0,因此第四列需要3):

    colname = df.columns[pos]
    

    样品:

    df = pd.DataFrame({'A':[1,2,3],
                       'B':[4,5,6],
                       'C':[7,8,9],
                       'D':[1,3,5],
                       'E':[5,3,6],
                       'F':[7,4,3]})
    
    print (df)
       A  B  C  D  E  F
    0  1  4  7  1  5  7
    1  2  5  8  3  3  4
    2  3  6  9  5  6  3
    
    pos = 3
    colname = df.columns[pos]
    print (colname)
    D
    

    pos = [3,5]
    colname = df.columns[pos]
    print (colname)
    Index(['D', 'F'], dtype='object')
    


知识点
面圈网VIP题库

面圈网VIP题库全新上线,海量真题题库资源。 90大类考试,超10万份考试真题开放下载啦

去下载看看