从CSV文件列创建字典

Question

从CSV文件列创建字典

4

我是一个对Python中的字典概念不熟悉的新手。我有一个包含多列的CSV文件，我想创建一个字典，使得键从第一列中获取，值从第二列中获取，并且为这两列的所有行创建一个键:值对。代码如下：

    if __name__=="__main__":
reader = csv.reader(open("file.csv", "rb"))
for rows in reader:
        k = rows[0]
        v = rows[1]
        mydict = {k:v}
print (mydict)

问题：返回结果只适用于第一列和第二列中的“最后”或“最底部”行，即{'12654'：'18790'}。我希望字典包含所有100行的前两列以这种格式。该怎么做？我可以在前两列的行号上运行一些循环来实现吗……我不知道如何做。

- user899714

你的意思是键从第一行中取出吗？ - Hod - Monica's Army

这段代码从最后一行获取键（最后一行的两列），但我希望它按顺序从第一行到最后一行覆盖文件中每一行的两列。 - user899714

2个回答

5

你正在创建一个新的字典并在每次迭代时覆盖旧字典。@develerx的答案解决了这个问题。我只是想指出一种更简单的方法，使用字典推导式：

假设csv文件包含两列。

if __name__=="__main__":
    reader = csv.reader(open("file.csv", "rb"))
    my_dict = {k: v for k, v in reader}
    print mydict

如果您正在使用旧版本（我认为早于2.7），则无法使用字典推导式，只需使用dict函数即可：

my_dict = dict((k, v) for k, v in reader)

编辑：我刚刚想到，my_dict = dict(reader) 也可以工作。

- utdemir

你可以使用第二个或第三个。 - utdemir

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- develerx · Accepted Answer

if __name__=="__main__":
    mydict = {}
    reader = csv.reader(open("file.csv", "rb"))
    for rows in reader:
        k = rows[0]
        v = rows[1]
        mydict[k] = v
    print mydict

在这里：

mydict = {k:v}

您在每次迭代中都创建了新的字典，之前的数据已经丢失。

更新：

您可以像这样做：

mydict = {}
L = [(1, 2), (2, 4), (1, 3), (3, 2), (3, 4)]
for el in L:
    k, v = el
    if not k in mydict:
        mydict[k] = [v]
    else:
        mydict[k].append(v)

print mydict

>>> 
{1: [2, 3], 2: [4], 3: [2, 4]}

这样，相同键的每个值都将被存储。

您的代码将是：

if __name__=="__main__":
    mydict = {}
    reader = csv.reader(open("file.csv", "rb"))
    for i, rows in enumerate(reader):
        if i == 0: continue
        k = rows[0]
        v = rows[1]
        if not k in mydict:
            mydict[k] = [v]
        else:
            mydict[k].append(v)

    print mydict

更新2：您的意思是？

for k, v in mydict.items():
    print "%s: %s" % (k, v)

>>>
1: [2, 3]
2: [4]
3: [2, 4]

更新3：

这个应该有效：

```html

这应该有效：

```

if __name__=="__main__":
        mydict = {}
        reader = csv.reader(open("file.csv", "rb"))
        for i, rows in enumerate(reader):
            if i == 0: continue
            k = rows[0]
            v = rows[1]
            if not k in mydict:
                mydict[k] = [v]
            else:
                mydict[k].append(v)

        print mydict