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))

从1d数组中的值开始切片2d数组

1 个答案: