如何索引列表中的3个最高值?

时间:2012-04-14 08:18:07

标签: python

所以我有这两个列表:

score = [350, 914, 569, 223, 947, 284, 567, 333, 697, 245, 227, 785, 120, 794, 343, 773, 293, 995]
name = [Ryan, Stacy, Jenna, Peter, Sophie, Bryan, Cole, Andrea, Emily, Blake, Mike, Stephan, Rob, Eliza, Heather, Daniel, Elisabeth, Samantha]

我必须找到3个最高分,并将这些分数与名单中的各自得分者进行统计,以便我可以有新的名单

top3score = [947, 995, 914]
top3name = [Sophie, Samantha, Stacy]

我正在考虑索引最高分,将它们添加到列表中,而不是使用索引将这些分数与名称相符。

我的问题是如何索引列表中的3个最高值? 然后,我如何使用索引在名称列表中查找得分者名称,以便我可以将它们附加到top3name列表中?

2 个答案:

答案 0 :(得分:24)

我认为这样做会

sorted(zip(score, name), reverse=True)[:3]

所以你明白发生了什么:

zip:将iterables作为参数并从每个iterable中获取一个元素,将它们放在元组中。

所以:

>>> zip(score, name)
[(350, 'Ryan'), (914, 'Stacy'), (569, 'Jenna'), (223, 'Peter'), (947, 'Sophie'), (284, 'Bryan'), (567, 'Cole'), (333, 'Andrea'), (697, 'Emily'), (245, 'Blake'), (227, 'Mike'), (785, 'Stephan'), (120, 'Rob'), (794, 'Eliza'), (343, 'Heather'), (773, 'Daniel'), (293, 'Elisabeth'), (995, 'Samantha')]

sorted:将对数据进行排序。默认情况下,元组元素在0索引中的元素上排序,因此在这种情况下得分。 Reverse = True将首先按降序排序。

最后,[:3]是切片符号,说给我从开头到第三个元素的所有元素。这也可以写成[0:3]

答案 1 :(得分:15)

如果您只对前三名感兴趣,那就是heapq.nlargest

>>> heapq.nlargest(3, zip(score, name))
[(995, 'Samantha'), (947, 'Sophie'), (914, 'Stacy')]

来自official doc

  

heapq.nlargest(n, iterable, key=None)

     
    

返回一个列表,其中包含iterable定义的数据集中的n个最大元素。 key(如果提供)指定一个参数的函数,该函数用于从iterable中的每个元素中提取比较键:key=str.lower相当于:sorted(iterable, key=key, reverse=True)[:n]

  

效果通知:

  

后两个[nlargestnsmallest]对较小的 n 值表现最佳。对于较大的值,使用sorted()函数更有效。此外,在n==1时,使用内置的min()max()函数会更有效。