python多项式相加
时间: 2023-12-06 16:38:37 浏览: 222
以下是Python实现多项式相加的示例代码:
```python
def add_poly(poly1, poly2):
result = {}
for exp, coef in poly1.items():
result[exp] = coef + result.get(exp,0)
for exp, coef in poly2.items():
result[exp] = coef + result.get(exp, 0)
return {exp: coef for exp, coef in result.items() if coef != 0}
```
上述代码中,我们定义了一个名为`add_poly`的函数,该函数接受两个字典类型的参数`poly1`和`poly2`,分别表示两个多项式。函数返回一个字典类型的结果,表示两个多项式相加的结果。
函数的实现过程如下:
1. 首先,我们创建一个空字典`result`,用于存储相加后的多项式。
2. 然后,我们遍历`poly1`字典中的每一项,将指数和系数分别存储到`exp`和`coef`变量中。
3. 接着,我们将`coef`加上`result`字典中指数为`exp`的项的系数,如果该项不存在,则默认系数为0。
4. 然后,我们遍历`poly2`字典中的每一项,将指数和系数分别存储到`exp`和`coef`变量中。
5. 接着,我们将`coef`加上`result`字典中指数为`exp`的项的系数,如果该项不存在,则默认系数为0。
6. 最后,我们使用字典推导式,将`result`字典中系数为0的项删除,并返回结果。
以下是一个示例,展示如何使用上述函数计算两个多项式的和:
```python
poly1 = {2: 3, 1: 4, 0: 1}
poly2 = {3: 2, 2: 1, 1: 3, 0: 1}
result = add_poly(poly1, poly2)
print(result) # 输出:{3: 2, 2: 4, 1: 7, 0: 2}
```
阅读全文