Pandas:如何通过索引获取列名
您可以使用以下方法通过 pandas 中的索引位置获取列名:
方法一:通过索引位置获取列名
#get column name in index position 2 colname = df. columns [ 2 ]
方法二:通过索引位置获取多个列名
#get column names in index positions 2 and 4 colname = df. columns [[ 2 , 4 ]]
以下示例展示了如何在实践中使用以下 pandas DataFrame 的每种方法:
import pandas as pd #createDataFrame df = pd. DataFrame ({' team ': ['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H'], ' points ': [18, 22, 19, 14, 14, 11, 20, 28], ' assists ': [5, 7, 7, 9, 12, 9, 9, 4], ' rebounds ': [11, 8, 10, 6, 6, 5, 9, 12], ' steals ': [4, 3, 3, 2, 5, 4, 3, 8], ' blocks ': [1, 0, 0, 3, 2, 2, 1, 5]}) #view DataFrame print (df) team points assists rebounds steals blocks 0 A 18 5 11 4 1 1 B 22 7 8 3 0 2 C 19 7 10 3 0 3 D 14 9 6 2 3 4 E 14 12 6 5 2 5 F 11 9 5 4 2 6 G 20 9 9 3 1 7:28 4 12 8 5
示例1:通过索引位置获取列名
以下代码显示如何获取 DataFrame 索引位置 2 处的列名称:
#get column name for column in index position 2 colname = df. columns [ 2 ] #display column name print (colname) assists
索引位置2处的列名是辅助的。
注意:Python中的列索引值从0开始。因此,索引位置 2 处的列将是 DataFrame 的第三列,名为Assists 。
示例2:通过索引位置获取多个列名
以下代码显示如何获取 DataFrame 索引位置 2 和 4 处的列的列名称:
#get column names in index positions 2 and 4 colname = df. columns [[ 2 , 4 ]] #display column names print (colname) Index(['assists', 'steals'], dtype='object')
从结果我们可以看出:
- 索引位置2处的列名是辅助的。
- 索引位置 4 处的列名称是Stees 。
注意:在此示例中,我们选择按索引位置获取两列的列名称,但您可以使用类似的语法来获取任意数量的列的列名称。
其他资源
以下教程解释了如何在 pandas 中执行其他常见任务:
如何从 Pandas DataFrame 获取单元格值
如何重命名 Pandas DataFrame 中的索引
如何在 Pandas 中按名称对列进行排序