使用enumerate和takewhile返回python中的列表

如何解决使用enumerate和takewhile返回python中的列表

我正在尝试定义一种方法，该方法使用enumerate（）和takewhile（）返回斐波纳契生成器的前n个元素的列表。我的代码进入无限循环，并且从不检查takewhile（）中的条件以停止迭代。我知道islice（）容易得多，但是我只需要使用枚举和获取时间来定义方法。

from itertools import takewhile


def fibonacci_unbounded():
    (a,b) = (0,1)
    while True:
        yield a
        (a,b) = (b,a + b)


def take_e(n,gen):
    fib_list = []
    for (i,elem) in enumerate(gen):
        if takewhile(i < n,gen):
            fib_list.append(elem)
        else:
            return fib_list


fibonacci = fibonacci_unbounded
n = 8

print(take_e(n,fibonacci()))

解决方法

这是一种有效的方法。我希望找到一种无需在逻辑中添加索引的方法，以便您可以测试返回的“ nth”项：

def fibonacci_unbounded():
    (a,b) = (0,1)
    while True:
        yield a
        (a,b) = (b,a + b)

def take_e(n,gen):
    return [x[1] for x in takewhile(lambda x: x[0] < n,enumerate(gen()))]

print(take_e(9,fibonacci_unbounded))

结果：

[0,1,2,3,5,8,13,21]

如果我不必使用takewhile，它将更加简单。我以前从未使用过它，所以我不知道使用它是否有特殊技巧。只需将所需的计数传递到基本斐波那契发生器中，并使其在适当的时间停止就容易了。

, 当使用zip迭代器时，

enumerate的工作原理与range相似。好的方面是zip在最短的迭代上停止。因此，抛弃takewhile并使用在正确位置停止的枚举器。

def fibonacci_unbounded():
    (a,gen):
    return list((elem) for _,elem in zip(range(n),gen))

fibonacci = fibonacci_unbounded
n = 8
print(take_e(n,fibonacci()))

如果需要使用takewhile和enumerate，则应从枚举中获取值，直到满足条件为止。

from itertools import takewhile

def fibonacci_unbounded():
    (a,gen):
    return list(val for _,val in takewhile(
        lambda i_val: i_val[0] < n,enumerate(gen)))

fibonacci = fibonacci_unbounded
n = 8
print(take_e(n,fibonacci()))