Python で最大または最小の N 要素を見つけるための実装コード

この記事では、Python で最大または最小の N 要素を見つけるための実装コードを紹介します。一定の参考値があります。必要な友人は参照できます。お役に立てれば幸いです。

1. 要件

特定のセット内の最大または最小の N 要素を見つけたいです

2.解決策

heapq モジュールには 2 つの関数があります: nlargest() と nsmallest()

コード:

import heapq
nums=[1,444,66,77,34,67,2,6,8,2,4,9,556]
print(heapq.nlargest(3,nums))
print(heapq.nsmallest(3,nums))

結果:

[556, 444, 77]
[1, 2, 2]

これら 2 つの各関数パラメータ キーを受け入れることができるため、より複雑なデータ構造を操作できるようになります。

コード:

import heapq
portfolio=[
    {'name':'IBM','shares':100,'price':91.1},
    {'name':'AAPL','shares':50,'price':543.22},
    {'name':'FB','shares':200,'price':21.09},
    {'name':'HPQ','shares':35,'price':31.75},
    {'name':'YHOO','shares':45,'price':16.35},
]
cheap=heapq.nsmallest(3,portfolio,key=lambda s:s['price'])
expensive=heapq.nlargest(3,portfolio,key=lambda s:s['price'])
print(cheap)
print(expensive)

結果:

[{'name': 'YHOO', 'shares': 45, 'price': 16.35}, {'name': 'FB', 'shares': 200, 'price': 21.09}, {'name': 'HPQ', 'shares': 35, 'price': 31.75}]
[{'name': 'AAPL', 'shares': 50, 'price': 543.22}, {'name': 'IBM', 'shares': 100, 'price': 91.1}, {'name': 'HPQ', 'shares': 35, 'price': 31.75}]

単純な場合は、最小または最大の要素 (N=1) の場合、min() と max() を使用する方が高速になります。

以上がPython で最大または最小の N 要素を見つけるための実装コードの詳細内容です。詳細については、PHP 中国語 Web サイトの他の関連記事を参照してください。