在Python数组中连续查找最小值的索引

时间:2019-03-11 18:36:15

标签: python

我似乎不知道该怎么做-我是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())

但是我似乎无法获得我想要的答案。任何帮助将不胜感激,谢谢

1 个答案:

答案 0 :(得分:1)

使用numpy argmin()

np.argmin(a, axis=1)

其中a是您的numpy数组。