如何通过列表从具有特定格式的其他列表中添加多个值的字典

时间:2012-12-05 05:50:12

标签: python python-3.x

我目前遇到一个我无法正确思考的问题

我的情况是我以特定格式读取文本文件

(捕食者)吃(猎物)

我试图做的是将它放入字典中,但有些情况下有多行。

(捕食者)吃(猎物)

同一只捕食者出现吃不同的猎物。

到目前为止,这就是它的样子......

import sys


predpraydic={}#Establish universial dictionary for predator and prey
openFile = open(sys.argv[1], "rt") # open the file

data = openFile.read() # read the file
data = data.rstrip('\n') #removes the empty line ahead of the last line of the file
predpraylist = data.split('\n') #splits the read file into a list by the new line character




for items in range (0, len(predpraylist)): #loop for every item in the list in attempt to split the values and give a list of lists that contains 2 values for every list, predator and prey
    predpraylist[items]=predpraylist[items].split("eats") #split "eats" to retrive the two values
    for predpray in range (0, 2): #loop for the 2 values in the list
        predpraylist[items][predpray]=predpraylist[items][predpray].strip() #removes the empty space caued by splitting the two values
for items in range (0, len(predpraylist)
    if 


for items in range (0, len(predpraylist)): # Loop in attempt to place these the listed items into a dictionary with a key of the predator to a list of prey
    predpraydic[predpraylist[items][0]] = predpraylist[items][1]

print(predpraydic)  
openFile.close() 

正如您所看到的,我只是将格式转储到我尝试转换为字典的列表中。

但是这个方法只接受一个键的值。我想要的东西有两件事,比如

狮子吃斑马 狮子吃狗

有一个

的字典 狮子:['斑马','狗']

我想不出这样做的方法。任何帮助将不胜感激。

1 个答案:

答案 0 :(得分:2)

有两种合理的方法可以创建包含您添加的列表的字典,而不是单个项目。第一种是在添加新值之前检查现有值。第二种是使用更复杂的数据结构,在必要时负责创建列表。

以下是第一种方法的快速示例:

predpreydic = {}

with open(sys.argv[1]) as f:
    for line in f:
        pred, eats, prey = line.split() # splits on whitespace, so three values
        if pred in predpreydic:
            predpreydic[pred].append(prey)
        else:
            predpreydic[pred] = [prey]

第一种方法的变体替换了if / else块,并在字典上稍微进行了一些细微的方法调用:

        predpreydic.setdefault(pred, []).append(prey)

setdefault方法将predpredic[pred]设置为空列表(如果它尚不存在),然后返回值(新的空列表或先前的现有列表)。它的工作方式与问题的另一种方法非常相似,接下来就是这样。

我提到的第二种方法涉及collections模块(Python标准库的一部分)中的the defaultdict class。这是一个字典,只要您请求一个尚不存在的密钥,它就会创建一个新的默认值。要按需创建值,它会使用您在首次创建defaultdict时提供的工厂函数。

以下是您的程序使用它的样子:

from collections import defaultdict

predpreydic = defaultdict(list) # the "list" constructor is our factory function

with open(sys.argv[1]) as f:
    for line in f:
        pred, eats, prey = line.split()
        predpreydic[pred].append(prey) #lists are created automatically as needed