Python：使用带有自定义迭代类对象的参数的函数，给出了不想要的结果

Question

我正在尝试为迭代自定义我自己的类，并尝试将其插入计算中：

class Iteration:
    def __init__(self, array):
        self.array = array

    def __pow__(self, power, modulo=None):
        new_array = list()
        for i in self.array:
            new_array.append(i ** power)
        return new_array

    def __len__(self):
        return len(self.array)

    def __getitem__(self, indices):
        return self.array[indices]


def mul(x):
    return x ** 2 + 3 * x ** 3


it = Iteration([1, 2, 3])

print(mul(2))   #=> 28
print(mul(it))  #=> [1, 4, 9, 1, 8, 27, 1, 8, 27, 1, 8, 27]

为什么mul（it）合并了重载结果？我该如何解决？我想要：print（mul（it））＃=> [4，28，90]

Answer 1

您的__pow__返回一个列表，而不是Iteration实例。 +和*操作是列表操作，列表将+和*实现为串联和重复。

[[1, 4, 9] + 3 * [1, 8, 27]重复[1, 8, 27] 3次以获得[1, 8, 27, 1, 8, 27, 1, 8, 27]，然后将[1, 4, 9]和[1, 8, 27, 1, 8, 27, 1, 8, 27]连接在一起。

您需要从Iteration返回一个__pow__实例，并且还需要实现__add__和__mul__，而不仅仅是__pow__。在使用它时，您可能还需要实现__str__或__repr__，因此在打印时可以看到Iteration对象包装的数据。