在NumPy中,广播(broadcasting)指的是不同形状的数组之间进行算术运算的规则。当两个数组的形状不同时,如果满足一些特定的条件,NumPy将自动地对它们进行广播以使得它们的形状相同。
广播的规则如下:
-
当两个数组的形状长度不同时,在较短的数组的前面加上若干个1,直到长度与较长的数组相同。
-
如果两个数组的形状在任何一个维度上不同且不同维度的长度不同,那么在进行广播之前,NumPy会比较它们的各维度长度。当满足以下任一条件时,认为这两个数组在该维度上是兼容的:
-
两个数组在这个维度上的长度相等
-
其中一个数组在这个维度上长度为1
-
如果两个数组在所有维度上都兼容,它们可以一起使用。
-
对于任何维度,其中一个数组长度为1,另一个数组长度大于1,第一个数组的元素在这个维度上重复,以匹配第二个数组的长度。
举个例子,假设有两个数组a和b,它们的形状分别为(3, 2)和(2,),那么可以使用广播将它们进行相加,具体如下:
a = np.array([[1, 2], [3, 4], [5, 6]])
b = np.array([1, 2])
c = a + b
输出结果为:array([[2, 4], [4, 6], [6, 8]])
在这个例子中,数组a的形状为(3, 2),数组b的形状为(2,),但是它们可以通过广播规则相加,产生形状为(3, 2)的结果数组c。具体来说,数组b被扩展为形状为(1, 2)的数组,然后复制成形状为(3, 2)的数组,与数组a进行相加得到结果数组c。
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:NumPy数组的广播是什么意思? - Python技术站