python - 具有对角矩阵的点积，无需创建完整矩阵

Question

我想计算两个矩阵的点积，其中一个矩阵是对角矩阵。但是，我不想使用 np.diag 或 np.diagflat 来创建完整矩阵，而是使用直接用对角线值填充的一维数组.有什么方法或 numpy 操作可以用来解决这类问题吗？

x = np.arange(9).reshape(3,3)
y = np.arange(3) # diagonal elements
z = np.dot(x, np.diag(y))

我正在寻找的解决方案应该没有 np.diag

z = x ??? y

Answer 1

直接将 ndarray 乘以你的矢量就可以了。 Numpy 方便地假设您想要将 x 的第 n 列乘以 y 的第 n 个元素。

x = np.random.random((5, 5)
y = np.random.random(5)

diagonal_y = np.diag(y)
z = np.dot(x, diagonal_y)
np.allclose(z, x * y)  # Will return True