在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

这应该可以解决问题：

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

在这种情况下，argmin不会像您期望的那样返回列表。

Answer 2

也许

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

它会给[0,5,7]。

Answer 3

我认为这是最简单的方法，虽然它不使用任何花哨的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)