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

Question

对 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}

Answer 1

您的 __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)