Python 实现 tail()

zltzlt

Python 实现 tail()

语法

tail(iterable, n)

描述

以生成器的形式返回 iterable 的后 n 个元素。如果 iterable 中的元素个数 < n，则返回 iterable 中的所有元素。

参数

参数	描述
iterable	可迭代对象。
n	必须是正整数。

返回值

生成器。

例子

>>> tail(range(10), 8)
[2, 3, 4, 5, 6, 7, 8, 9]
>>> tail(range(10), 3)
[7, 8, 9]
>>> tail(range(10), 10)
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
>>> tail(range(10), 15)
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

代码

from _collections import deque as _deque


def tail(iterable, n):
    """
    tail(iterable, n) -> [generator object]
    以列表的形式返回可迭代对象的后 n 个元素
    """
    return list(_deque(iterable, n))