Loading

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

posted @ 2021-03-28 17:30  克豪  阅读(329)  评论(0)    收藏  举报