博客

获取Pandas数据框架的某一列值的列表

  • Post category:Python

获取Pandas数据框架（DataFrame）的某一列值的列表，可以通过pandas库提供的类似Python字典键值对的方式来操作。以下是详细的攻略：

  1. 读取数据框架：首先，需要通过pandas库的read_csv方法或其他文件读取方法，将数据文件读入一个数据框架。例如我们有一个csv文件，包含学生id、姓名、年龄、性别等信息，我们可以通过如下代码读取文件并将其转化为数据框架：
import pandas as pd

df = pd.read_csv('students.csv')
  1. 查看数据框架：可以使用head方法来查看前几行数据框架的内容，以验证数据框架已经正确读入。例如，我们可以使用如下代码查看前5行数据：
print(df.head())
  1. 获取某一列：使用类似Python字典键值对的方式，获取数据框架的某个列。例如，我们要获取学生的年龄，可以使用如下代码获取年龄列：
ages = df['age']
  1. 转换为列表：将获取的年龄列转换为列表类型，方便后续操作。转换可以使用tolist方法，例如：
age_list = ages.tolist()
  1. 输出结果：最后，我们可以使用print方法输出年龄列表：
print(age_list)

由以上步骤得到的完整代码如下：

import pandas as pd

df = pd.read_csv('students.csv')
print(df.head())

ages = df['age']
age_list = ages.tolist()

print(age_list)

其中，students.csv是一个包括学生信息的csv文件，列名为id、name、age、gender等。上述代码将年龄列值以列表形式输出。

Tags: