python - 将 numpy 分类数据映射到 numpy 向量

Question

我有一个看起来像的 numpy 数组：

my_arr = array([[0., 0., 0., 0., 1., 0.],
   [0., 1., 0., 0., 0., 0.],
   [0., 0., 0., 1., 0., 0.],
   [0., 0., 0., 0., 1., 0.],
   [1., 0., 0., 0., 0., 0.],
   [0., 0., 0., 0., 1., 0.],
   [0., 1., 0., 0., 0., 0.],
   [0., 0., 0., 0., 1., 0.],
   ...
   ...]

我想返回一个向量，该向量将为 my_arr 的每个向量包含值为 1 的条目索引。我该怎么做？

标签： pythonnumpy

Answer 1

你用np.argmax()它。

inds = np.argmax(my_arr, axis=1)
# array([4, 1, 3, 4, 0, 4, 1, 4])