如何在 Python 中从 CSV 文件创建字典？ [复制]

Question

我有一个 CSV 文件，有两列：“id”和“quantity”。该文件有一个标题，其中包含提到的两列（id 和数量），以及从第二行开始的值。

例如，文件如下所示：

id, quantity
1234,100
2345,200
3456,300
4567,400

我想从 csv 创建一个 Python dict，它在字典中读取上面的 CSV，以便我以后可以轻松检查字典中是否有特定的 id，如果存在，值是什么（数量）。

有人知道怎么做吗？

Answer 1

是的，只需使用内置的csv 模块：

import csv
import itertools

with open('data.csv') as f:
    reader = csv.reader(f)
    next(reader) # skip the header
    stream = ((int(x), int(y)) for x,y in reader)
    my_dict = dict(stream)