如何在 Python 中将字典写入 Excel

Question

我在 python 中有以下字典，表示从 - 到距离矩阵。

graph = {'A':{'A':0,'B':6,'C':INF,'D':6,'E':7},

         'B':{'A':INF,'B':0,'C':5,'D':INF,'E':INF},

         'C':{'A':INF,'B':INF,'C':0,'D':9,'E':3},

         'D':{'A':INF,'B':INF,'C':9,'D':0,'E':7},

         'E':{'A':INF,'B':4,'C':INF,'D':INF,'E':0}

         }

是否可以将此矩阵输出到 excel 或 csv 文件中，使其具有以下格式？我已经研究过使用 csv.writer 和 csv.DictWriter 但无法产生所需的输出。

Answer 1

您可以从该字典创建一个熊猫数据框，然后保存到 CSV 或 Excel：

import pandas as pd
df = pd.DataFrame(graph).T  # transpose to look just like the sheet above
df.to_csv('file.csv')
df.to_excel('file.xls')