從1d數組中的值開始切片2d數組

Question

具有一個任意的二維數組，例如零，以及一個索引數組：

z = np.zeros((5,5))
ix = np.array([1,4,2,3,0])

如何從1d數組指定的列開始添加1 ，以獲得：

array([[0, 1, 1, 1, 1],
       [0, 0, 0, 0, 1],
       [0, 0, 1, 1, 1],
       [0, 0, 0, 1, 1],
       [1, 1, 1, 1, 1]])

我無法找到一種使用numpy的簡單方法。

Answer 1

一種方法是-

In [50]: ncols = 5

In [51]: (ix[:,None] <= np.arange(ncols)).view('i1')
Out[51]: 
array([[0, 1, 1, 1, 1],
       [0, 0, 0, 0, 1],
       [0, 0, 1, 1, 1],
       [0, 0, 0, 1, 1],
       [1, 1, 1, 1, 1]], dtype=int8)

如果必須添加到現有數組z

z += (ix[:,None] <= np.arange(ncols))