2016-11-08 69 views
0

我正在Jupyter Notebook中使用python 3。python,極小值數組索引

可以想象我有:

import numpy as np 

q = np.array([5, 2, 6, 7, 2]) 

我現在想找到數組的最小值的指標;在這種情況下,數字1和4

我試着使用:

np.argmin(q) 

這給:

1 

不幸的是np.argmin()的時候,只有一個最小的纔有效。我如何找到另一個最小值的索引?

回答

1

使用np.where()

np.where(q == q.min())[0] 

結果:

array([1, 4])