python - to_dict() 在值周围创建括号

Question

我正在尝试从我的 pandas DataFrame 创建默认字典，但 to_dict() 方法在我要写入的列的值周围创建了不需要的方括号。示例代码如下:

# Create DF
my_df = pd.DataFrame({'numbers': (1, 2, 3, 4, 5), 'letters': ('a', 'b', 'c', 'd', 'e')})

# Create dictionary from the DF
my_dict = my_df.set_index('numbers').T.to_dict('list')

# Create collections dictionary
my_collections_dict = collections.defaultdict(int, my_dict)

结果:

defaultdict(int, {1: ['a'], 2: ['b'], 3: ['c'], 4: ['d'], 5: ['e']})

我想要的是:

defaultdict(int, {1: 'a', 2: 'b', 3: 'c', 4: 'd', 5: 'e'})

如何获取“纯”列值？

Answer 1

您不需要转置您的框架，您可以选择您的列并执行以下操作:

my_dict = my_df.set_index('numbers')['letters'].to_dict()

---

如果你想在你的字典中使用多个列，它会占用额外的一行，但你可以使用:

my_dict = my_df.set_index('numbers').to_dict(orient='index')
my_dict = {k: list(v.values()) for k, v in my_dict.items()}