python - Pythonで指定された配列の最小値のインデックスを見つける

Question

配列で発生する複数の最小値のインデックスを見つける必要があります。私はよく知られてnp.argminいますが、配列内の最初の最小値のインデックスが得られます。例えば。

a = np.array([1,2,3,4,5,1,6,1])    
print np.argmin(a)

これで 0 が得られますが、期待していたのは 0,5,7 です。

ありがとう！

Answer 1

多分

mymin = np.min(a)
min_positions = [i for i, x in enumerate(a) if x == mymin]

[0,5,7]になります。

Answer 2

派手なnumpy関数は使用しませんが、これが最も簡単な方法だと思います

a       = np.array([1,2,3,4,5,1,6,1])                                        
min_val = a.min()                                                            

print "min_val = {0}".format(min_val)                                        

# Find all of them                                                           
min_idxs = [idx for idx, val in enumerate(a) if val == min_val]              
print "min_idxs = {0}".format(min_idxs)