python列表求最小值索引,在Python中查找給定數組中的最小值索引

I need to find the index of more than one minimum values that occur in an array. I am pretty known with np.argmin but it gives me the index of very first minimum value in a array. For example.

我需要找到數組中出現的多個最小值的索引。我很熟悉np.argmin，但它給了我一個數組中第一個最小值的索引。例如。

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

print np.argmin(a)

This gives me 0, instead I am expecting, 0,5,7.

這給了我0，而不是我期待的，0,5,7。

Thanks!

謝謝！

3 个解决方案

#1

21

This should do the trick:

這應該是訣竅：

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

print np.where(a == a.min())

argmin doesn't return a list like you expect it to in this case.

在這種情況下，argmin不會像您期望的那樣返回列表。

#2

3

Maybe

也許

mymin = np.min(a)

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

It will give [0,5,7].

它會給[0,5,7]。

#3

1

I think this would be the easiest way, although it doesn't use any fancy numpy function

我認為這將是最簡單的方法，雖然它不使用任何花哨的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)