时间戳

首页 > 解决方案 > 如何使用 csv 文件创建字典的字典

python - 如何使用 csv 文件创建字典的字典

问题描述

我想创建一个字典，键是csv文件的第一行，值是{第一列：对应列到行}的字典 例如，test.csv

Header1, Header2, Header3
A,       1,       10
B,       2,       20
C,       3,       30

我正在使用data.drop(['Header1'], 1)drop Header1。

我希望得到的 dict 是：

{'Header2':{A:1, B:2, C:3},'Header3':{A:10, B:20, C:30}}

我应该使用csv.DictReader吗？还是 numpy 数组？

标签： pythonpandascsvdictionary

解决方案

使用熊猫，

df = pd.read_csv(testcsv, index_col=['Header1'], sep=',\s+')
df.to_dict()

输出：

{'Header2': {'A': 1, 'B': 2, 'C': 3}, 'Header3': {'A': 10, 'B': 20, 'C': 30}}

推荐阅读