18 lines
699 B
Markdown
18 lines
699 B
Markdown
---
|
|
title: max_n
|
|
tags: math
|
|
---
|
|
Returns the `n` maximum elements from the provided list. If `n` is greater than or equal to the provided list's length, then return the original list(sorted in descending order).
|
|
|
|
Use `list.sort()` combined with the `deepcopy` function from the inbuilt `copy` module to create a shallow clone of the list and sort it in ascending order and then use `list.reverse()` reverse it to make it descending order. Use `[:n]` to get the specified number of elements. Omit the second argument, `n`, to get a one-element list
|
|
|
|
```py
|
|
def max_n(lst, n=1, reverse=True):
|
|
return sorted(lst, reverse=reverse)[:n]
|
|
```
|
|
|
|
```py
|
|
max_n([1, 2, 3]) # [3]
|
|
max_n([1, 2, 3], 2) # [3,2]
|
|
```
|