30-seconds-of-code/snippets/max-n.md

title, type, tags, cover, dateModified

title	type	tags	cover	dateModified
N max elements	snippet	list math	red-succulent	2020-11-02T19:28:27+02:00

Returns the n maximum elements from the provided list.

• Use sorted() to sort the list.
• Use slice notation to get the specified number of elements.
• Omit the second argument, n, to get a one-element list.
• If n is greater than or equal to the provided list's length, then return the original list (sorted in descending order).

def max_n(lst, n = 1):
  return sorted(lst, reverse = True)[:n]

max_n([1, 2, 3]) # [3]
max_n([1, 2, 3], 2) # [3, 2]