如何在Python中的矩陣和向量之間使用逐元素乘法

Question

以下是我的輸入：

a = np.array([[1,1],[1,1],[1,1]])
b = np.array([2,3,4])

我想使用乘以b的列：

result = np.array([[2,2],[3,3],[4,4]])

我怎么能得到這個結果？

Answer 1

您可以轉置->廣播->轉置：

>>> (a.T * b).T
array([[2, 2],
       [3, 3],
       [4, 4]])