30秒学会 Python 片段 · 2019年6月27日

30秒学会 Python 片段 – take_right.md


title: take_right
tags: list,beginner

Returns a list with n elements removed from the end.

Use slice notation to create a slice of the list with n elements taken from the end.

代码实现

def take_right(itr, n = 1):
  return itr[-n:]

使用样例

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