从Pandas数据框架中的行创建一个列表

  • Post category:Python

步骤如下:

  1. 导入pandas模块。
import pandas as pd
  1. 创建一个Pandas数据框架。

假设我们有以下的数据框架。

班级 姓名 年龄
1 小明 10
1 小红 11
2 小张 9
2 小李 10

使用以下代码创建数据框架。

df = pd.DataFrame({'班级': [1, 1, 2, 2], '姓名': ['小明', '小红', '小张', '小李'], '年龄': [10, 11, 9, 10]})
  1. 从数据框架中选取行并将其转化为列表。

选取“班级”为1的行并将其转化为列表。

class_one_list = df[df['班级']==1].values.tolist()
  1. 打印转化后的列表。
print(class_one_list)

输出结果:

[[1, '小明', 10], [1, '小红', 11]]

以上就是从Pandas数据框架中的行创建一个列表的完整攻略,希望对你有所帮助。