pandas data frame转换成list
时间: 2023-09-04 17:03:53 浏览: 85
将pandas的DataFrame转换为list的方法有很多。下面是几种常见的转换方法:
方法一:使用values属性
可以使用DataFrame的values属性将其转换为一个NumPy数组,然后再将数组转换为list。示例代码如下:
```
import pandas as pd
# 创建一个示例DataFrame
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})
# 将DataFrame转换为list
df_list = df.values.tolist()
print(df_list)
```
上述代码会输出结果`[[1, 4], [2, 5], [3, 6]]`,其中每个子列表表示DataFrame的一行数据。
方法二:使用to_dict()方法
DataFrame对象还提供了to_dict()方法,可以将DataFrame按列转换为一个字典,再将字典转换为list。示例代码如下:
```
import pandas as pd
# 创建一个示例DataFrame
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4,5,6]})
# 将DataFrame转换为list
df_list = df.to_dict('records')
print(df_list)
```
上述代码会输出结果`[{'A': 1, 'B': 4}, {'A': 2, 'B': 5}, {'A': 3, 'B': 6}]`,其中每个字典表示DataFrame的一行数据。
这两种方法都可以将DataFrame转换为list,选择哪种方法取决于具体的需求和数据结构。
阅读全文