python - 重载 + 以支持元组

Question

我希望能够用 python 编写这样的东西:

a = (1, 2)
b = (3, 4)
c = a + b # c would be (4, 6)
d = 3 * b # d would be (9, 12)

我知道您可以重载运算符以处理自定义类，但有没有办法重载运算符以处理对？

当然有这样的解决方案

c = tuple([x+y for x, y in zip(a, b)])

确实有效，但是撇开性能不谈，它们不如重载 + 运算符那么漂亮。

当然可以定义add 和mul 函数，例如

def add((x1, y1), (x2, y2)):
    return (x1 + x2, y1 + y2)

def mul(a, (x, y)):
    return (a * x, a * y)

但仍然能够编写 q * b + r 而不是 add(times(q, b), r) 会更好。

想法？

编辑:附带一提，我意识到由于 + 目前映射到元组连接，重新定义它可能是不明智的，即使它是可能的。该问题仍然适用于 - 例如 =)

Answer 1

与 Ruby 不同，您无法更改 Python 中内置类型的行为。您所能做的就是从内置类型派生 创建一个新类型。不过，文字仍会创建内置类型。

可能你能得到的最好的是

class T(tuple):
    def __add__(self, other):
        return T(x + y for x, y in zip(self, other))
    def __rmul__(self, other):
        return T(other * x for x in self)
a = T((1, 2))
b = T((3, 4))
c = a + b # c would be (4, 6)
d = 3 * b # d would be (9, 12)