如何将每个 numpy 列中的所有非零元素分配给大小与列数相同的数组中的值？

Question

So that's a bit of a mouthful. But here's what I'm looking to do:

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

I hardly think there is a faster/neater answer for this. Writing for others to find it helpful. As @Mark mentioned in the comments, you can find non-zero elements by a>0 and multiplying it into b will broadcast b to a 's shape by repeating rows and multiply element-wise:

output = (a > 0) * b

Another way would be:

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