Python類：使用添加一個數字，每一個元素的矢量__add__

Question

假設我有一個向量（A_1，A_2，A_3，...，A_N）

，我想添加一個數字的每個元素我的矢量在Python類中使用__add__。

使得：

A =矢量（7,4，2）

打印一個+ 3

將輸出：（10，7,5）

請幫助。我的代碼是什麼問題？

非常感謝。

Answer 1

class Vector(): 

    def __init__(self, *args): 
     self.args = args 

    def __add__(self, other): 
     return tuple(map(lambda x:x+other,self.args)) 

a = Vector(1,2,3,4,5) 
print a+3 

Answer 2

這裏有一個簡單的實現：

class vector(object): 

    def __init__(self, *args): 
     self.args = args 

    def __add__(self, other): 
     if not isinstance(other, int): 
      raise TypeError("Cannot add {} with {}".format(type(self), type(other))) 
     return vector(*(arg + other for arg in self.args)) 

    def __repr__(self): 
     return "vector({})".format(", ".join(map(str, self.args))) 


a = vector(7, 4, 2) 

print a + 3 # (10, 7, 5) 

輸出：

vector(10, 7, 5) 

我以爲你(10, 7, 4)本來是(10, 7, 5)