如何將每個 numpy 列中的所有非零元素分配給大小與列數相同的數組中的值？

Question

所以這有點拗口。 但這是我想要做的：

b = np.array([7,8,2,3])

a = np.array([[1, 1, 0, 1],
              [0, 0, 1, 1],
              [0, 1, 1, 0]])

*** The Magic Happens ***

array([[7, 8, 0, 3],
       [0, 0, 2, 3],
       [0, 8, 2, 0]])

Answer 1

我幾乎不認為有一個更快/更整潔的答案。 為其他人寫作以發現它有幫助。 正如評論中提到的@Mark，您可以通過a>0找到非零元素，並將其乘以b將通過重復行並將元素相乘來廣播b到a的形狀：

output = (a > 0) * b

另一種方法是：

a[a>0] = np.tile(b,(a.shape[0],1))[a>0]