如何从值为集合的字典创建Key、Value分列存储的Pandas DataFrame

实现方法 #

首先注意不要使用dict作为变量名，它是Python内置关键字，会覆盖原生字典类的功能，我们先将原字典重命名为src_dict，再通过以下代码生成目标DataFrame：

import pandas as pd

# 定义原字典
src_dict = {'A':{1,2,4}, 'B':{5,6}}

# 构造DataFrame
df = pd.DataFrame(src_dict.items(), columns=['Key', 'Value'])

# 如果需要Key列的值带单引号，可加上下面这行
df['Key'] = df['Key'].apply(lambda x: f"'{x}'")

打印df即可得到你需要的基础格式：

Key      Value
0   'A'  {1, 2, 4}
1   'B'     {5, 6}

如果需要Value列的集合完全无空格匹配示例的{1,2,4}格式，可以额外增加一行代码处理：

df['Value'] = df['Value'].astype(str).str.replace(' ', '')

内容的提问来源于stack exchange，提问作者Tai Huang