将Numpy三维阵列乘以1D阵列

2024-04-25 11:57:26 发布

您现在位置:Python中文网/ 问答频道 /正文

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

1D_array[d]*2D_array

最后我得到了一个数组,它看起来像:

[[
[1,1]
[1,1]]
[
[2,2]
[2,2]]
[
[3,3]
[3,3]]]

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

我已经尝试了一段时间了,不管我做什么,我都不能得到这个结果,只是主题的变化。我该如何正确地进行这项工作?在

谢谢你的帮助。在


Tags: 主题npones公司数组array试图用