Question

我定义了一个名为DictList的类。 DictList类表示一个dict列表，其中一些键出现在多个dict中。它有一个参数：它匹配一个或多个参数（必须是字典）。

class DictList:
    def __init__(self,*args):
        self.n = args

喜欢：

A = DictList({'a':1, 'b':2, 'c':3}, {'c':5, 'd':7, 'e':16}, {'e':25, 'f':29)

现在我想在这个类中定义一个 add 方法。它会将两个DictList一起添加或添加一个DictList和一个字典。

示例：

>>> A1 = DictList(dict(a=1,b=2), dict(b=12,c=13)) 
>>> A2 = DictList(dict(a='hi',b='hello'), dict(b='good',c='nice'))

>>> A1+A2 
DictList({'a': 1, 'b': 2}, {'b': 12, 'c': 13}, {'a': 'hi', 'b': 'hello'}, {'b': 'good', 'c': 'nice'})

>>> A2 + A1
DictList( {'a': 'hi', 'b': 'hello'}, {'b': good', 'c': 'nice'},{'a': 1, 'b': 2}, {'b': 12, 'c': 13})

我的想法是创建一个新的DictList，其中包含一个字典列表，其中包含两个DictLists中的所有dicts或一个DictList和一个dict。

def __add__(self,new):
        if type(new) == DictList:
            print(DictList( all dicts from self and new ))
        if type(new) == dict:
            print(DictList(all dicts from self and new))

但是我不知道如何在两个DictList中获取每个dict并将它们全部放入新的DictList中。我该怎么办？

Answer 1

使用已加入的DictList参数列表返回新的self.n。如果您还想添加常规字典，请检查实例的类型。您可能希望定义__radd__来处理dict + DictList而不是DictList + dict。

class DictList:
    def __init__(self,*args):
        self.n = args

    def __add__(self,new):
        L = list(self.n)
        if isinstance(new,DictList):
            L.extend(new.n)
        elif isinstance(new,dict):
            L.append(new)
        else:
            raise TypeError('Must be instance of DictList or dict')
        return DictList(*L)

    def __radd__(self,new):
        return self.__add__(new)

    def __repr__(self):
        return 'DictList'+repr(self.n)

演示：

>>> d = DictList(dict(a=1,b=2,c=3),dict(a=2,b=3,e=4))
>>> e = DictList(dict(a=4,c=2,e=3))
>>> d+e
DictList({'b': 2, 'c': 3, 'a': 1}, {'b': 3, 'a': 2, 'e': 4}, {'c': 2, 'a': 4, 'e': 3})
>>> d+dict(x=1,y=2)
DictList({'b': 2, 'c': 3, 'a': 1}, {'b': 3, 'a': 2, 'e': 4}, {'x': 1, 'y': 2})
>>> dict(x=1,y=2)+d
DictList({'b': 2, 'c': 3, 'a': 1}, {'b': 3, 'a': 2, 'e': 4}, {'x': 1, 'y': 2})
>>> d+5
Traceback (most recent call last):
  File "<interactive input>", line 1, in <module>
  File "C:\Users\metolone\Desktop\x.py", line 12, in __add__
    raise TypeError('Must be instance of DictList or dict')
TypeError: Must be instance of DictList or dict

如何在python中的特定类中构建add方法？

1 个答案:

如何在python中的特定类中构建__add__方法？

1 个答案:

如何在python中的特定类中构建add方法？