如何将两个numpy数组对应位置的数字相加?
时间: 2023-06-10 19:08:40 浏览: 339
python中利用numpy.array()实现俩个数值列表的对应相加方法
5星 · 资源好评率100%
可以使用numpy的加法运算符("+")来实现对应位置的数字相加。示例如下:
``` python
import numpy as np
a = np.array([1, 2, 3])
b = np.array([4, 5, 6])
c = a + b
print(c) # 输出 [5 7 9]
```
如果两个数组的形状不同,可以使用numpy的广播功能进行自动扩展,使它们的形状相同后再进行加法运算。示例如下:
``` python
import numpy as np
a = np.array([[1, 2], [3, 4]])
b = np.array([10, 20])
c = a + b
print(c) # 输出 [[11 22], [13 24]]
```
在上面的示例中,数组b被自动扩展为[[10, 20], [10, 20]],使它的形状与数组a相同,然后两个数组对应位置的数字相加得到数组c。
阅读全文