从标量和矩阵形成逐元素列表

Question

我有一个零维

numpy

标量

s

和一个二维

numpy

矩阵

m

。我想形成一个向量矩阵，其中

m

的所有元素都与

s

配对，如下例所示：

import numpy as np

s = np.asarray(5)

m = np.asarray([[1,2],[3,4]])

# Result should be as follows

array([[[5, 1],
        [5, 2]],

       [[5, 3],
        [5, 4]]])

换句话说，我想在

np.asarray([s, m])

的最低级别按元素对操作

m

进行向量化。对于

m

内的任何多维数组

numpy

，是否有一种明显的方法可以做到这一点？

我确信这在某个地方，但我很难用语言表达它，也找不到它。如果您能找到它，请随时将我重定向到那里。

Answer 1

一个可能的解决方案：

np.stack([np.broadcast_to(s, m.shape), m], axis=-1)