2024-04-18 06:32:26 发布
网友
有人能帮我了解一下广播是怎么工作的吗np.哪里()功能?在
x = np.arange(9.).reshape(3, 3) np.where(x < 5, x, -1) # Note: broadcasting. array([[ 0., 1., 2.], [ 3., 4., -1.], [-1., -1., -1.]])
让我们看看各个部分
x = np.arange(9).reshape(3, 3) >>> x array([[0, 1, 2], [3, 4, 5], [6, 7, 8]])
请注意,x < 5生成布尔数组:
x < 5
将此插入np.where:
np.where
>>> np.where(x < 5, x, -1) array([[ 0, 1, 2], [ 3, 4, -1], [-1, -1, -1]])
请注意,-1已被广播以匹配x < 5的维度:
-1
array([[-1, -1, -1], [-1, -1, -1], [-1, -1, -1]])
由于x已经具有正确的维度,因此不需要任何广播。在
x
让我们看看各个部分
请注意,
^{pr2}$x < 5
生成布尔数组:将此插入
np.where
:请注意,
-1
已被广播以匹配x < 5
的维度:由于
x
已经具有正确的维度,因此不需要任何广播。在相关问题 更多 >
编程相关推荐