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
转载请注明原文地址:https://ipadbbs.8miu.com/read-47750.html