在Python中,将包含关键字对的列表添加到字典的最简单方法是什么?

时间:2015-03-19 03:22:04

标签: python csv python-3.x dictionary

我在Python中有一个家庭作业问题。
我在Linux上使用Python 3.4.0版 设计文档声明我要使用内置函数读取CSV文件,这些函数指定为names.dat,格式如下:

name:name2, name:name3, name2:name4, name3:name5\n (etc)

然后我将这些关键字对添加到字典中,这是我坚持的部分。
到目前为止我的代码是:

dictionary = dict()
database = open('names.dat', 'r')
data = database.read()
data = data.rstrip('\n')
data = data.split(',')
for item in range(len(data)):
    dictionary.update(data[item-1])

我的想法是,如果我有一个格式为“name:name2”的列表元素,并且我将该元素作为参数调用字典更新函数,它将正确映射到字典中的关键字对。 /> 但是,情况并非如此,因为我在运行此脚本时遇到此错误:

File "MyName.py", line 7, in <module>
    dictionary.update(data[item-1])
ValueError: dictionary update sequence element #0 has length 1; 2 is required

ThisThis似乎相似,但我觉得这是一个不同的问题,需要单独回复。
我在这里做错了什么,我该如何解决呢? 有更简单的方法吗?

2 个答案:

答案 0 :(得分:5)

如果你想从给定的csv创建一个精确的数据集,@ Paulo Scardine有一个很好的答案。如果要根据键组合值,可以使用:

changes = {}
with open('test.csv', 'r') as f:
    for row in f:
        for e in row.rstrip('\n').split(", ") : #split lines by column
            print (e) #just to show what is being generated here
            (k,v) = e.split(":") #split further into key, value pairs
            changes.setdefault(k, []).append(v)
            #creates empty list if new key, adds value to list

print (changes)

数据看起来像:

{'name3': ['name5'], 'name2': ['name4', 'name6', 'name5'], 'name1': ['name', 'name4'], 'name': ['name2', 'name3']}

这可以进一步简化,但我认为这给了一个好的例子,有人可以学习。

编辑:在@Paulo Scardine评论后添加了setdefault方法

答案 1 :(得分:3)

试试这个:

data = []
with open('names.dat') as database:
    for line in database:
        if line.strip():  # skip blank lines
            data.append(
                dict(i.split(":") for i in line.rstrip('\n').split(","))
            )

如果您的文件是:

name:name2,name:name3,name2:name4,name3:name5
name:name2,name:name3,name2:name4,name3:name5
name:name2,name:name3,name2:name4,name3:name5
name:name2,name:name3,name2:name4,name3:name5

data将是:

[{'name': 'name3', 'name2': 'name4', 'name3': 'name5'},
 {'name': 'name3', 'name2': 'name4', 'name3': 'name5'},
 {'name': 'name3', 'name2': 'name4', 'name3': 'name5'},
 {'name': 'name3', 'name2': 'name4', 'name3': 'name5'}]

也许你想要一个dict list而不是list dict

data = {}
with open('names.dat') as database:
    for line in database:
        if line.strip():  # skip blank lines
            for k, v in (i.split(":") for i in line.rstrip('\n').split(",")):
                data.setdefault(k, []).append(v)

由于:

{'name': [ 'name2', 'name3', 'name2', 'name3', 'name2', 'name3', 'name2', 'name3'],
 'name2': ['name4', 'name4', 'name4', 'name4'],
 'name3': ['name5', 'name5', 'name5', 'name5']}