小编jgi*_*ist的帖子

我目前有一个数组中最小值的索引数组.

它看起来像这样:

[[0],
 [1],
 [2],
 [1],
 [0]]

(最大指数为3)

我想要的是一个如下所示的数组:

[[1, 0, 0]
 [0, 1, 0]
 [0, 0, 1]
 [0, 1, 0]
 [1, 0, 0]]

1在最小列中.

在numpy中有一个简单的方法吗？