21 lines
748 B
Markdown
21 lines
748 B
Markdown
### max_n
|
|
|
|
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 array
|
|
|
|
```python
|
|
from copy import deepcopy
|
|
|
|
|
|
def max_n(arr, n=1):
|
|
numbers = deepcopy(arr)
|
|
numbers.sort()
|
|
numbers.reverse()
|
|
return numbers[:n]
|
|
```
|
|
|
|
```python
|
|
max_n([1, 2, 3]) # [3]
|
|
max_n([1, 2, 3], 2) # [3,2]
|
|
``` |