我目前遇到一个我无法正确思考的问题
我的情况是我以特定格式读取文本文件
(捕食者)吃(猎物)
我试图做的是将它放入字典中,但有些情况下有多行。
(捕食者)吃(猎物)
同一只捕食者出现吃不同的猎物。
到目前为止,这就是它的样子......
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()
正如您所看到的,我只是将格式转储到我尝试转换为字典的列表中。
但是这个方法只接受一个键的值。我想要的东西有两件事,比如
狮子吃斑马 狮子吃狗有一个
的字典 狮子:['斑马','狗']我想不出这样做的方法。任何帮助将不胜感激。
答案 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