我有一个列表,其中包含多次出现的最小元素,例如
a = [1,2,1,1,4,5,6]
我希望 Python 返回列表中存在的元素1
和所有索引。1
我尝试使用
min_index, min_value = min(enumerate(a), key=operator.itemgetter(1))
它只给出1
首先出现的索引。
我会这样做:
minimum = min(a)
indices = [i for i, v in enumerate(a) if v == minimum]
确定最小元素,然后将其与列表中的其他元素进行检查。
def locate_min(a):
smallest = min(a)
return smallest, [index for index, element in enumerate(a)
if smallest == element]
这将返回一个元组 (min_element, [location, location, ...])。如果我理解正确,这就是我认为你想要的。对于您的示例:
>>> locate_min([1, 2, 1, 1, 4, 5, 6])
(1, [0, 2, 3])
此示例使用列表推导。如果你不熟悉这个,它大致相当于下面的 for-loop 版本。(使用第一个版本,这只是为了帮助您理解它是如何工作的)
def locate_min(a):
min_indicies = []
smallest = min(a)
for index, element in enumerate(a):
if smallest == element: # check if this element is the minimum_value
min_indicies.append(index) # add the index to the list if it is
return smallest, min_indicies