python - Pandas 数据框到有序字典

Question

有一篇文章将 Pandas 数据框转换为字典以供进一步处理。

执行此操作的代码是:

df = pd.read_excel(open('data/file.xlsx', 'rb'), sheetname="Sheet1")
dict = df.set_index('id').T.to_dict('dict')

这会产生类似这样的结果:{column -> {index -> value}}

有没有一种快速的方法来代替这个 {column -> {index -> value}} 得到这个:OrderedDict(column, value) 作为返回值？

目前，我正在使用从 pandas 生成的字典，并将这些值一个一个地分配给有序字典。这不是最佳方式，因为顺序被打乱了

示例输入:像这样的 Excel 文件:

Unique_id | column1 | column2 | column3 | column 4
1         | 3       | 4       | 43      | 90
2         | 54      | 6       | 43      | 54

并且输出应该是一个有序字典，如下所示:

{1:[3,4,43,90], 2:[54,6,43,54]}

Answer 1

您可以通过使用带有 Unique_id 列中的键的 OrderedDict 以所需的顺序获取字典。以下应作为说明:

from collections import OrderedDict

# Get the unordered dictionary
unordered_dict = df.set_index('Unique_id').T.to_dict('list')

 # Then order it
ordered_dict = OrderedDict((k,unordered_dict.get(k)) for k in df.Unique_id)
# OrderedDict([(1, [3, 4, 43, 90]), (2, [54, 6, 43, 54])])

谢谢!