Question

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

例如，文件看起来像这样：

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

我想从csv创建一个Python字典，该字典在字典中读取上面的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)

如何在Python中从CSV文件创建字典？

1 个答案: