Numpy技巧(一)之----numpy数组广播

numpy数组广播

numpy数组广播的一般规则是在列,在维度按照右对齐后,两个数组是兼容的。这里兼容的意思是说,要么在对齐维度上相当具有相同的维数,要么必须有一个为一。而且为一的会被广播为不为一的。

举个例子。下面这两组数字为两个数组在各个维度的维数,按上面规则广播,有,

A      (4d array):  8 x 1 x 6 x 1
B      (3d array):      7 x 1 x 5
Result (4d array):  8 x 7 x 6 x 5

在一维数组的情况下,默认情况下是行数变多。因为列数对齐,列数一般不为一,此时行数在右边第二维。因此,广播之后行数变多。因此,默认情况下,好像是把一维数组当成了一个行向量,在不断的往下复制。

A      (2d array):  5 x 4
B      (1d array):      4
Result (2d array):  5 x 4

当然,列数为一,按规则两维扩展,此时好像一个标量复制20遍,

A      (2d array):  5 x 4
B      (1d array):      1
Result (2d array):  5 x 4

要想列数变多。应该把它变成 $n imes 1 $ 二维数组的形式。这时列数为一广播之后,列数变多。这时$n imes 1 $ 二维数组才会看成一个列向量,不断地往右复制。如,

A      (2d array):  5 x 4
B      (2d array):  5 x 1
Result (2d array):  5 x 4
原文地址:https://www.cnblogs.com/qizhien/p/11583466.html