用函数实现两个类两元素的加减
时间: 2024-02-05 19:13:34 浏览: 42
假设有两个类ClassA和ClassB,每个类有两个元素a和b,可以按照以下方式定义函数实现加法和减法操作:
```python
class ClassA:
def __init__(self, a, b):
self.a = a
self.b = b
class ClassB:
def __init__(self, a, b):
self.a = a
self.b = b
def add(class1, class2):
result_a = class1.a + class2.a
result_b = class1.b + class2.b
return ClassA(result_a, result_b)
def subtract(class1, class2):
result_a = class1.a - class2.a
result_b = class1.b - class2.b
return ClassB(result_a, result_b)
```
在上面的代码中,我们首先定义了两个类ClassA和ClassB,每个类有两个元素a和b。接着我们定义了两个函数add和subtract,add函数将两个类的元素进行相加并返回结果,subtract函数将两个类的元素进行相减并返回结果。
你可以按照以下方式使用这些函数:
```python
class1 = ClassA(1, 2)
class2 = ClassA(3, 4)
result_add = add(class1, class2)
print(result_add.a, result_add.b) # 输出 4 6
class3 = ClassB(5, 6)
class4 = ClassB(3, 2)
result_subtract = subtract(class3, class4)
print(result_subtract.a, result_subtract.b) # 输出 2 4
```
在上面的代码中,我们创建了两个ClassA类型的对象class1和class2,以及两个ClassB类型的对象class3和class4,分别进行加法和减法操作,并输出结果。