广播机制:通过扩展实现数组运算
来自CloudWiki
numpy数组的广播机制
- 注意numpy里数组之间的运算和线性代数中的矩阵运算是不一样的
- 同纬度和形状的数组之间就是同位置的元素两两匹配运算的结果
import numpy as np
a=np.array([[ 1., 2., 1.],
[ 1., 2., 1.]])
b=np.array([[ 1., 2., 3.],
[ 1., 2., 3.]])
print(a.shape,b.shape)
>>>(2, 3) (2, 3)
print(a+b)
>>>array([[ 2., 4., 4.],
[ 2., 4., 4.]])
print(a*b)
>>>array([[ 1., 4., 3.],
[ 1., 4., 3.]])
print(a-b)
>>>array([[ 0., 0., -2.],
[ 0., 0., -2.]])
print(a**b)
>>>array([[ 1., 4., 1.],
[ 1., 4., 1.]])