如何为一个键添加一组多个值?

时间:2017-06-07 01:29:48

标签: python-3.x class dictionary set

我创建了一个基本上是爱好书的课程。这本书可以通过两种方法访问,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

1 个答案:

答案 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来填充。