python - 查找numpy数组中一行中最小值的索引

Question

我似乎不知道如何做到这一点 - 我是 Python 和这种数组工作的新手。我有一个很大的数组，说:

array([[119., 323.,  42., 277., 401.],
       [122., 326.,  39., 278., 10.],
       [125., 329.,  36., 12., 407.],
       ...,
       [308., 314., 469., 188., 266.],
       [308., 314., 469., 188., 266.],
       [308., 314., 469., 188., 266.]])

我想找到每行中最小值的列索引。例如，前 3 行将给出 [2, 4, 3....]。例如，我尝试过 .min() 和 np.where():

np.where(array == array.min())

但我似乎无法得到我正在寻找的答案。任何帮助将不胜感激，谢谢

Answer 1

使用numpy argmin() :

np.argmin(a, axis=1)

其中 a 是您的 numpy 数组。