N个最小元素

从提供的列表中返回n个最小元素。

  • 使用sorted()对列表进行排序。
  • 使用切片表示法获取指定数量的元素。
  • 如果省略第二个参数n,则返回一个包含一个元素的列表。
  • 如果n大于或等于提供的列表的长度,则返回原始列表(按升序排序)。
def min_n(lst, n = 1):
  return sorted(lst, reverse = False)[:n]

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