Multiplying Numpy 3D arrays by 1D arrays

不打扰是莪最后的温柔 提交于 2019-12-07 17:13:57

问题


I am trying to multiply a 3D array by a 1D array, such that each 2D array along the 3rd (depth: d) dimension is calculated like:

1D_array[d]*2D_array

And I end up with an array that looks like, say:

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

Which would be the result of correctly multiplying np.ones((3,2,2)) with [1,2,3].

I've been trying for some time now and whatever I seem to do I can't end up with this result, just variations on the theme. How do I correctly go about doing this?

Thanks for any help.

来源:https://stackoverflow.com/questions/14513222/multiplying-numpy-3d-arrays-by-1d-arrays

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!