要获取Pandas数据框架中指定列的列表,可以使用DataFrame的loc和iloc方法,通过选择行和列的位置来获取指定列。
步骤如下:
- 首先,导入Pandas库和创建数据框架
import pandas as pd
df = pd.DataFrame({'name': ['Alice', 'Bob', 'Charlie'], 'age': [25, 32, 18], 'gender': ['F', 'M', 'F']})
- 通过loc方法获取指定列
age_list = df.loc[:, 'age'].tolist()
print(age_list)
输出:
[25, 32, 18]
- 通过iloc方法获取指定列
gender_list = df.iloc[:, 2].tolist()
print(gender_list)
输出:
['F', 'M', 'F']
在上述代码中,:,:
选择了所有的行,,age
以列名称作为索引选择了指定列。:2
以列的位置(从0开始)作为索引选择了指定列。
同时,使用tolist()函数将获取的列值转换为列表类型。
上述代码中的输出结果均为列表类型。其中第一个列表是年龄列,第二个列表是性别列。
注意:在实际的操作中,可以根据需要选择使用loc还是iloc方法,两个方法区别是loc方法使用的是行和列的名称,而iloc方法使用的是行和列的位置。