如何读取 2 列 csv 文件并创建字典？

Question

例如给出以下 csv

ID, type
1 , A
2 , B
3 , C

它应该生成一个看起来像这样的字典

{'1':A, '2':B, '3':C}

到目前为止，这是我所拥有的，但它将整个列关联到 1 个字典中

import csv

reader = csv.DictReader(open('TIS_annotation.csv'))
result = {}

for row in reader:
    for column, value in row.iteritems():
        result.setdefault(column, []).append(value)
print result

Answer 1

当您遍历每个rowin 时reader，该row变量包含在字典中创建新条目所需的所有信息。你可以简单地写

for row in reader:
    result[row['ID']] = row[' type']

制作你想要的字典。

Answer 2

它比你想象的要简单：

import csv

with open('TIS_annotation.csv') as f:
    next(f)  # Skip the header
    reader = csv.reader(f, skipinitialspace=True)
    result = dict(reader)
    print result

输出：

{'1 ': 'A', '3 ': 'C', '2 ': 'B'}

基本上，reader产生一系列行，每行有两个元素，将其输入dict并制作。