将 df 转为 dict - Python

如果你有一个 pandas 的 DataFrame 数据结构，并且想要将它转换为 Python 的字典数据类型，那么这篇文章就是为你准备的。

方法一：使用 to_dict() 函数

pandas.DataFrame.to_dict() 函数可以将 DataFrame 转换为字典类型。以下是一个简单的示例，展示如何使用 to_dict() 函数转换 DataFrame 为字典：

import pandas as pd

df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})

# 将 DataFrame 转换为字典
my_dict = df.to_dict()

print(my_dict)

输出：

{'A': {0: 1, 1: 2, 2: 3}, 'B': {0: 4, 1: 5, 2: 6}}

方法二：使用 zip() 函数

另一种将 DataFrame 转换为字典的方法是使用 zip() 函数。以下是一个使用 zip() 函数的示例：

import pandas as pd

df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})
my_dict = dict(zip(df.columns, [df[col].tolist() for col in df.columns]))

print(my_dict)

输出：

{'A': [1, 2, 3], 'B': [4, 5, 6]}

总结

这篇文章介绍了两种将 pandas 的 DataFrame 转换为 Python 字典的方法。我们可以使用内置的 to_dict() 函数或使用 zip() 函数来将 DataFrame 转换为字典类型。两种方法都非常简单，但是需要您知道如何操作 pandas 和 Python 的数据结构。