python - 遍历一个可迭代对象，如何使用next()获取前一个值和下一个值？

Question

def f(iterable):
    i = iter(iterable)
    int_list = []
    n2 = next(i)
    while True:
        n1, n2, n3 = n2, next(i), next(i)
        print('n1', n1)
        print('n2,', n2)
        print('n3', n3)
        if n2 > n1 and n2> n3:
            int_list.append(n2)
    return int_list

在这种情况下，可迭代对象是一个列表。我想检查整数是否大于前一个整数和下一个整数。但是我不知道如何分配正确的值以使用 next() 方法进行检查

Answer 1

您可以创建以下生成器:

def f(iterable):
    i = iter(iterable)
    n1, n2, n3 = next(i), next(i), next(i)
    while True:
        if n2 > max(n1, n3):
            yield n2
        n1, n2, n3 = n2, n3, next(i)

然后像这样测试:

>>> list(f([1, 4, 3, 8, 6]))
[4, 8]