dataframe转换为字典
时间: 2023-07-15 14:12:08 浏览: 83
Python 将DataFrame数据转成字典 Python源码
可以使用Pandas库中的`to_dict()`方法将DataFrame转换为字典。该方法支持将DataFrame转换为多种不同的格式,其中一种就是字典。
以下是示例代码:
```python
import pandas as pd
# 创建一个DataFrame对象
df = pd.DataFrame({'a': [1, 2, 3], 'b': ['foo', 'bar', 'baz']})
# 将DataFrame转换为字典
d = df.to_dict()
print(d)
```
输出结果如下:
```
{'a': {0: 1, 1: 2, 2: 3}, 'b': {0: 'foo', 1: 'bar', 2: 'baz'}}
```
可以看到,`to_dict()`方法返回了一个字典,其中每个列名都对应一个字典,该字典的键是行索引,值是对应的数据。如果希望将列名作为键,可以使用`orient`参数指定转换方向。
例如,以下代码将以列名为键,将每一列转换为一个字典:
```python
d = df.to_dict(orient='dict')
print(d)
```
输出结果如下:
```
{'a': {0: 1, 1: 2, 2: 3}, 'b': {0: 'foo', 1: 'bar', 2: 'baz'}}
```
同样的,如果希望将行索引作为键,可以使用`orient`参数指定为`'index'`。
阅读全文