Python字典在第一个之后不添加后续键

时间:2017-09-21 01:20:52

标签: python algorithm dictionary data-structures

相当新的Python,我无法弄清楚这一点。我去添加一个字典的密钥,它添加它很好。我甚至可以使用新值更新相同的密钥,但是当我向字典添加第二个密钥时,它不会添加第二个密钥值对。

class CountedSet:
    def __init__(self):
        self.data = {}
    def __iadd__(self,other):
        if isinstance(other,int):
            self.data[other] = self.data.get(other, 0) + 1
            return self
        elif isinstance(other,CountedSet):
            #TODO::iterate through second countedSet and update self
            return self
    def __add__(self,obj):
        for key, value in obj.data.items():
            if len(self.data) == 0:
                self.data[key] = value
            elif self.data[key]:
                self.data[key] = self.data[key] + value
            else:
                self.data[key] = value
        return self
    def __getitem__(self,item):
        if item in self.data:
            return self.data.get(item)
        else:
            return None
    def __str__(self):
        for key, value in self.data.items():
            return("{%s,%s}" % (key,value))
a = CountedSet()
a += 17
a += 4
print(a)

当我希望看到{17,1} {4,1}

时,这只会输出{17,1}

1 个答案:

答案 0 :(得分:3)

你的__str__实现在for循环的第一次迭代中返回:

def __str__(self):
    for key, value in self.data.items():
        return("{%s,%s}" % (key,value)) # here

也许你想要这样的东西:

def __str__(self):
    return " ".join([{"{%s,%s}" % (k,v) for k, v in self.data.items()])

或者,没有理解:

def __str__(self):
    items = []
    for key, value in self.data.items():
        items.append("{%s,%s}" % (key,value))
    return ' '.join(items)