小编dsc*_*scm的帖子

将 Numpy 3D 数组乘以 1D 数组

我试图将一个 3D 数组乘以一个 1D 数组,这样沿着第 3 个(深度:d)维度的每个 2D 数组的计算方式如下:

1D_array[d]*2D_array

我最终得到一个看起来像这样的数组,比如:

[[
[1,1]
[1,1]]
[
[2,2]
[2,2]]
[
[3,3]
[3,3]]]
Run Code Online (Sandbox Code Playgroud)

这将是将 np.ones((3,2,2)) 与 [1,2,3] 正确相乘的结果。

我已经尝试了一段时间,无论我似乎做什么,我都无法得到这个结果,只是主题的变化。我该如何正确地执行此操作?

谢谢你的帮助。

python arrays numpy multidimensional-array array-broadcasting

6
推荐指数
1
解决办法
4161
查看次数