Python에서 가장 크거나 가장 작은 N 요소를 찾기 위한 구현 코드

이 기사는 Python에서 가장 큰 또는 가장 작은 N 요소를 찾는 구현 코드를 제공합니다. 여기에는 특정 참조 값이 있으므로 도움이 될 수 있습니다.

1. 요구 사항

특정 세트에서 가장 크거나 작은 N 요소를 찾고 싶습니다

2. 솔루션

heapq 모듈에는 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]

두 함수 모두 매개변수 키를 허용하여 더 복잡한 데이터 구조에서 작업할 수 있습니다.

코드:

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 중국어 웹사이트의 기타 관련 기사를 참조하세요!