在Python中查找给定数组中最小值的索引答案

【问题标题】：Find the index of minimum values in given array in Python在Python中查找给定数组中最小值的索引
【发布时间】：2013-11-02 00:50:09
【问题描述】：

我需要找到一个数组中出现的多个最小值的索引。我对np.argmin 非常熟悉，但它为我提供了数组中第一个最小值的索引。例如。

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

这给了我 0，而我期待的是 0,5,7。

谢谢！

【问题讨论】：

试试np.where(a == a.min())
Getting the index of the returned max or min item using max()/min() on a list 的可能重复项

标签： python arrays numpy

【解决方案1】：

也许

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

它会给出 [0,5,7]。

【讨论】：

不，它没有，因为您正在与最小值的索引进行比较，即 0。
已编辑，使用 min 而不是 argmin，抱歉。

【解决方案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)

【讨论】：

【解决方案3】：

这应该可以解决问题：

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

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

【讨论】：

不，它没有。此外，您正在覆盖一个内置函数min。
@BasSwinckels 你错过了我的编辑。正如您所指出的，常规 min 可以正常工作。