如何使用Python编写一个向量加法或减法计算器,并能够显示计算结果?
时间: 2024-12-10 12:29:13 浏览: 12
python实现计算器功能
在Python中,你可以创建一个简单的函数来执行向量加法或减法。首先,你需要定义一个类表示向量,包含一个元素列表。然后,你可以定义加法和减法的方法,它们接受另一个向量作为参数,通过对应位置元素相加或相减来完成操作。下面是一个例子:
```python
class Vector:
def __init__(self, *args):
self.elements = list(args)
def add(self, other):
if len(self.elements) != len(other.elements):
raise ValueError("Vectors must have the same dimension to be added.")
result = [a + b for a, b in zip(self.elements, other.elements)]
return Vector(*result)
def subtract(self, other):
if len(self.elements) != len(other.elements):
raise ValueError("Vectors must have the same dimension to be subtracted.")
result = [a - b for a, b in zip(self.elements, other.elements)]
return Vector(*result)
def display(self):
print(f"Vector: {self.elements}")
# 使用示例
vector1 = Vector(1, 2, 3)
vector2 = Vector(4, 5, 6)
sum_vector = vector1.add(vector2)
sum_vector.display() # 输出: Vector: [5, 7, 9]
difference_vector = vector1.subtract(vector2)
difference_vector.display() # 输出: Vector: [-3, -3, -3]
```
在这个例子中,`add` 和 `subtract` 方法会检查两个向量是否维度相同,如果不一致则抛出异常。`display` 方法用于打印向量内容。
阅读全文