dataframe的to_dict
时间: 2024-05-21 10:08:13 浏览: 9
DataFrame的to_dict()方法是用于将DataFrame对象转换为字典的方法。它可以将DataFrame的每一列转换为字典的键值对形式,其中列名作为键,列中的元素作为值。
to_dict()方法有几个参数可以控制转换的方式:
- orient:指定字典的排列方式,默认为'columns',表示以列为基准,每一列转换为一个字典。还可以选择'index',表示以行为基准,每一行转换为一个字典。
- into:指定字典值的类型,默认为dict,表示转换为普通的字典。还可以选择OrderedDict,表示转换为有序字典。
以下是一个示例:
```
import pandas as pd
data = {'Name': ['Tom', 'John', 'Emma'],
'Age': [25, 30, 28],
'City': ['New York', 'London', 'Paris']}
df = pd.DataFrame(data)
dict_columns = df.to_dict(orient='columns')
dict_index = df.to_dict(orient='index')
print(dict_columns)
print(dict_index)
```
输出结果:
```
{'Name': {0: 'Tom', 1: 'John', 2: 'Emma'}, 'Age': {0: 25, 1: 30, 2: 28}, 'City': {0: 'New York', 1: 'London', 2: 'Paris'}}
{0: {'Name': 'Tom', 'Age': 25, 'City': 'New York'}, 1: {'Name': 'John', 'Age': 30, 'City': 'London'}, 2: {'Name': 'Emma', 'Age': 28, 'City': 'Paris'}}
```