将向量与矩阵元素相乘[重复]

Question

我有一个 (n,1) np 数组，例如 array([1 2 3])，我想将元素与 np 矩阵 (n,m) 相乘，例如 array([[1 1 1], [2 2 2], [3 3 3]])，这样我会得到：

array([[1 1 1], [4 4 4], [9 9 9]])

我该怎么做？

我尝试过使用 np.multiply 和 np.dot。

Answer 1

重塑您的矢量，使其包含 3 行而不是 3 列：

v = np.array([1, 2, 3])
m = np.array([[1, 1, 1], [2, 2, 2], [3, 3, 3]])

u = v.reshape(*v.shape, 1)
u * m  # results in [[1, 1, 1], [4, 4, 4], [9, 9, 9]]