python peek函数

在 Python 中，没有内置的 peek() 函数。但是，可以通过使用一些技巧来实现一个类似于 peek() 的函数。下面是一种实现方式：

def peek(iterable):
    """Return the next item in the iterable without advancing the iterator."""
    try:
        peeked_item = next(iterable)
        iterable = chain([peeked_item], iterable)
    except StopIteration:
        return None
    return peeked_item

这个函数接受一个可迭代对象作为输入，并返回该对象的下一个元素，而不会使迭代器向前移动。在实现中，我们使用了 `next()` 函数和 `itertools.chain()` 函数。首先，我们使用 `next()` 函数获取下一个元素并保存在 `peeked_item` 变量中。然后，我们使用 `chain()` 函数将 `peeked_item` 和原始的可迭代对象连接起来，以便在下一次调用 `next()` 函数时，可以继续从原始可迭代对象中获取元素。如果可迭代对象已经被耗尽，我们返回 None。

使用 `peek()` 函数的示例：

from itertools import chain

my_list = [1, 2, 3, 4, 5]
iterable = iter(my_list)

print(peek(iterable))  # Output: 1
print(next(iterable))  # Output: 1
print(peek(iterable))  # Output: 2
print(next(iterable))  # Output: 2