Python Numpy中的广播机制(broadcasting)可以使代码的运行速度更快、减少代码数量以下是其加减乘除简单用法的例子。
1、加法:
import numpy as np
A = np.array([[1,2,3],
[1,3,5],
[2,4,6]])
B = np.array([100,200,300])
print("A=",A)
print("B=",B)
print("A+B=",A+B)

两个矩阵列相同,行不相同,则广播机制将行数少的矩阵扩展使得两个矩阵可以相加。
2、减法:
import numpy as np
A = np.array([[1,2,3],
[1,3,5],
[2,4,6]])
B = np.array([[100],[200],[300]])
print("A=",A)
print("B=",B)
print("B-A=",B-A)

同样的行相同,列不相同,则广播机制将列数少的矩阵扩展。
3、乘法:
import numpy as np
A = np.array([[1,2,3],
[1,3,5],
[2,4,6]

最低0.47元/天 解锁文章
2569

被折叠的 条评论
为什么被折叠?



