30秒学会 Python 片段 – Remove list elements from the end

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]

翻译自：https://www.30secondsofcode.org/python/s/take-right