如何将 DF 转换为字典

Question

这是我的代码：

import pandas as pd
import xlrd

p=pd.read_excel("D:\contacts.xlsx")
pp=p.set_index('Col-A').T
print(pp)
dic=pp.to_dict()

print(dic) 

我的DF：

    Col-A  Col-B
0      1     10
1      2     20
2      3     30
3      4     40
4      5     50
5      6     60
6      7     70
7      8     80
8      9     90
9     10    100

预期输出： {1:10,2:20,3:30,4:40,5:50}

我的输出看起来像：

Col-A  1   2   3   4   5 
Col-B  10  20  30  40  50  
{1: {'Col-B': 10}, 2: {'Col-B': 20}, 3: {'Col-B': 30}, 4: {'Col-B': 40}, 5: {'Col-B': 50}}

我怎样才能达到预期的输出？

Answer 1

首先set_index然后做to_dict并调用Col-B

df.set_index('Col-A').to_dict()['Col-B']

# {1: 10, 2: 20, 3: 30, 4: 40, 5: 50, 6: 60, 7: 70, 8: 80, 9: 90, 10: 100}

Answer 2

to_dict 方法接收一个参数“orient”，默认值为 dict。

您可以将此输出设置 orient 转换为“列表”。

df.to_dict('list')

您可以在此之后迭代以将您的列表平铺为单个值。

https://pandas.pydata.org/pandas-docs/version/0.21/generated/pandas.DataFrame.to_dict.html