获取Pandas数据框架的指定列的列表

  • Post category:Python

要获取Pandas数据框架中指定列的列表,可以使用DataFrame的loc和iloc方法,通过选择行和列的位置来获取指定列。

步骤如下:

  1. 首先,导入Pandas库和创建数据框架
import pandas as pd

df = pd.DataFrame({'name': ['Alice', 'Bob', 'Charlie'], 'age': [25, 32, 18], 'gender': ['F', 'M', 'F']})
  1. 通过loc方法获取指定列
age_list = df.loc[:, 'age'].tolist()
print(age_list)

输出:

[25, 32, 18]
  1. 通过iloc方法获取指定列
gender_list = df.iloc[:, 2].tolist()
print(gender_list)

输出:

['F', 'M', 'F']

在上述代码中,:,:选择了所有的行,,age以列名称作为索引选择了指定列。:2以列的位置(从0开始)作为索引选择了指定列。

同时,使用tolist()函数将获取的列值转换为列表类型。

上述代码中的输出结果均为列表类型。其中第一个列表是年龄列,第二个列表是性别列。

注意:在实际的操作中,可以根据需要选择使用loc还是iloc方法,两个方法区别是loc方法使用的是行和列的名称,而iloc方法使用的是行和列的位置。