我创建了一个基本上是爱好书的课程。这本书可以通过两种方法访问,enter(n,h)
取名并继续为该名称添加爱好(一个名称可以有多个爱好)。另一种方法返回特定名称的一组爱好。我的爱好书正在存储我插入一个名字的每个爱好。有人可以帮我修理吗?
class Hobby:
def __init__(self):
self.dic={}
self.hby=set()
def enter(self,n,h):
if n not in self.dic.items():
self.dic[n]=self.hby
for k in self.dic.items():
self.hby.add(h)
def lookup(self,n):
return self.dic[n]
我尝试运行以下案例
d = Hobby(); d.enter('Roj', 'soccer'); d.lookup('Roj')
{'soccer'}
d.enter('Max', 'reading'); d.lookup('Max')
{'reading', 'soccer'} #should return just reading
d.enter('Roj', 'music'); d.lookup('Roj')
{'reading', 'soccer','music'} #should return soccer and music
答案 0 :(得分:2)
为什么要在这里重新发明dict
?为什么使用一个单独的集合来总是添加值,并将其引用到每个键,以确保它始终在查找中返回相同的集合?
请勿重新发明轮子,请使用collections.defaultdict
:
import collections
d = collections.defaultdict(set)
d["Roj"].add("soccer")
d["Roj"]
# {'soccer'}
d["Max"].add("reading")
d["Max"]
# {'reading'}
d["Roj"].add("music")
d["Roj"]
# {'soccer', 'music'}
更新 - 如果您真的想通过自己的课程(在此之前,请观看Stop Writing Classes!),您可以这样做:
class Hobby(object):
def __init__(self):
self.container = {}
def enter(self, n, h):
if n not in self.container:
self.container[n] = {h}
else:
self.container[n].add(h)
def lookup(self, n):
return self.container.get(n, None)
d = Hobby()
d.enter("Roj", "soccer")
d.lookup("Roj")
# {'soccer'}
d.enter("Max", "reading")
d.lookup("Max")
# {'reading'}
d.enter("Roj", "music")
d.lookup("Roj")
# {'soccer', 'music'}
请注意,此处没有使用额外的设置 - 每个dict
密钥都有自己的set
来填充。