numpy不同维度的数组进行运算
numpy中不同维度的数组进行相加的时候,只要其中一个数组的维度是另一个数组的子维度,那么他们就可以进行运算,相当于把低维数组通过复制,升级到高维,然后再进行运算
比如:
a = np.arange(24).reshape(4,6)
a
array([[ 0, 1, 2, 3, 4, 5],
[ 6, 7, 8, 9, 10, 11],
[12, 13, 14, 15, 16, 17],
[18, 19, 20, 21, 22, 23]])
b = np.arange(6)
b
array([0, 1, 2, 3, 4, 5])
a+b
array([[ 0, 2, 4, 6, 8, 10],
[ 6, 8, 10, 12, 14, 16],
[12, 14, 16, 18, 20, 22],
[18, 20, 22, 24, 26, 28]])
相当于a的每一行都加上了b

浙公网安备 33010602011771号