在Python列表中查找最大值和索引?

23

我有一个类似于这样的Python列表:

[[12587961, 0.7777777777777778], [12587970, 0.5172413793103449], [12587979, 0.3968253968253968], [12587982, 0.88], [12587984, 0.8484848484848485], [12587992, 0.7777777777777778], [12587995, 0.8070175438596491], [12588015, 0.4358974358974359], [12588023, 0.8985507246376812], [12588037, 0.5555555555555555], [12588042, 0.9473684210526315]]

这个列表可能有上千个元素,我该如何根据子数组中的第二项获取列表中的最大值,并获取最大值的索引,该索引是子数组中的第一个元素,在Python中怎么做?


你是对的 @unwind。我认为他想要找到所有第二个元素具有最大值的元素,并基于第一个值找到其中的最大值。 - thefourtheye
很抱歉如果你们没明白,我的意思是我想要具有最大第二个值的子数组,第二个值指的是子数组中的[1]项。 - rksh
好的...我仍然很难理解。举个例子,比如一个四五个元素的列表会很有帮助。但我猜这个问题已经解决了。 - unwind
@rksh 如果有多个子数组具有最大的第二个值,该怎么办? - thefourtheye
5个回答

39
使用max函数及其key参数,仅使用列表中的第二个元素进行比较元素。
例如,
>>> data = [[12587961, 0.7777777777777778], [12587970, 0.5172413793103449], [12587979, 0.3968253968253968].... [12588042, 0.9473684210
526315]]
>>> max(data, key=lambda item: item[1])
[12588042, 0.9473684210526315]

现在,如果您只想获取第一个元素,那么您可以仅获取第一个元素,或者只需解包结果,如下所示。
>>> index, value = max(data, key=lambda item: item[1])
>>> index
12588042
>>> value
0.9473684210526315

编辑:如果您想找到所有具有最大值(第二个值)的元素中的最大索引(第一个值),则可以按照以下方式执行。
>>> _, max_value = max(data, key=lambda item: item[1])
>>> max(index for index, value in data if value == max_value)

你可以在单个迭代中完成相同的操作,像这样:
max_index = float("-inf")
max_value = float("-inf")

for index, value in data:
      if value > max_value:
          max_value = value
          max_index = index
      elif value == max_value:
          max_index = max(max_index, index)

谢谢,实际上我想要的是您在原始答案中提到的类似内容,即子数组中具有最大第二个值的子数组 :) 是否有一种方法可以获取具有最大值的n个数组,例如前10个最高值 :) - rksh
@rksh,你应该给出恰当的例子。这些话只会让人感到困惑。 - thefourtheye
好的,我来举个简单的例子。如果有一个数组 data = [1,2,5,7,6,8,3,9,4],我该如何获取其中三个最大值呢?在这种情况下,它们是 7,8,9 - rksh
将它们排序并切片结果列表。 - thefourtheye

5

使用带有键的 max 函数。

l = [[12587961, 0.7777777777777778], [12587970, 0.5172413793103449], [12587979, 0.3968253968253968], [12587982, 0.88], [12587984, 0.8484848484848485], [12587992, 0.7777777777777778], [12587995, 0.8070175438596491], [12588015, 0.4358974358974359], [12588023, 0.8985507246376812], [12588037, 0.5555555555555555], [12588042, 0.9473684210526315]]
max_sub = max(l, key=lambda x: x[1])
max_val = max_sub[1]
max_index = max_sub[0]

1
from operator import itemgetter

a = [[12587961, 0.7777777777777778], [12587970, 0.5172413793103449], [12587979, 0.3968253968253968], [12587982, 0.88], [12587984, 0.8484848484848485], [12587992, 0.7777777777777778], [12587995, 0.8070175438596491], [12588015, 0.4358974358974359], [12588023, 0.8985507246376812], [12588037, 0.5555555555555555], [12588042, 0.9473684210526315]]

max(a, key=itemgetter(1))[0]
// => 12588042

0
allData = [[12587961, 0.7777777777777778], [12587970, 0.5172413793103449], [12587979, 0.3968253968253968], [12587982, 0.88], [12587984, 0.8484848484848485], [12587992, 0.7777777777777778], [12587995, 0.8070175438596491], [12588015, 0.4358974358974359], [12588023, 0.8985507246376812], [12588037, 0.5555555555555555], [12588042, 0.9473684210526315]]

listOfSecondData = [i[1] for i in allData]
result = allData[listOfSecondData.index(max(listOfSecondData))][0]

print(result)
#Output: 12588042

-3

简单

list = [[12587961, 0.7777777777777778], [12587970, 0.5172413793103449], [12587979, 0.3968253968253968], [12587982, 0.88], [12587984, 0.8484848484848485], [12587992, 0.7777777777777778], [12587995, 0.8070175438596491], [12588015, 0.4358974358974359], [12588023, 0.8985507246376812], [12588037, 0.5555555555555555], [12588042, 0.9473684210526315]]
list2 = []

for x in list:
    list2.append(x[1])
print "index->" + str(list[list2.index(max(list2))][0])
print "max value->" + str(list[list2.index(max(list2))][1])

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接