Numpy-broadcast机制

Numpy中的广播机制是为了解决不同维度数组之间的算术问题。

广播流程：

当两个数组形状相同时，算术运算可以顺利进行：

import numpy as np
a = np.array([
    [0, 0, 0],
    [10, 20, 30],
    [40, 50, 60]
])
 
b = np.array([[1, 2, 3],
              [5, 1, 8],
              [2, 4, 8]])
print(a+b)

输出结果：

[[ 1  2  3]
 [15 21 38]
 [42 54 68]]

当两个数组形状不同时，Numpy数组会自动触发广播机制，可以认为时对其中一个维度较小的数组进行数据的拓展（将第一行的数据复制n次），从而生成两个相同维度的数组。

import numpy as np
a = np.array([
    [0, 0, 0],
    [10, 20, 30],
    [40, 50, 60]
])
 
b = np.array([1, 2, 3])
print(a+b)

输出结果：

[[ 1  2  3]
 [11 22 33]
 [41 52 63]]

当其中一个数组非一维数组，那就无法进行广播了：