如果index>len（list），如何按索引循环列表并从头开始？

Question

本质：

对于像['A', 'B', 'C']这样的列表，我试图构建或查找一个函数，该函数将在每次调用它时返回列表的下一个元素。在本例中，如果调用它超过三次，那么A将被返回。然后B等等

---

细节和我的尝试：

我有一个颜色列表['A', 'B', 'C']，我正在应用于不同的对象。每次运行代码时，对象的数量可能会有所不同。如果列表中的对象多于颜色，我希望第四个对象的颜色为'A'。因此，正如标题所说，我想循环遍历一个列表by和index，如果index > len(list)，则从头开始。下面的自定义函数looper将实现这一点

def looper(lst, runs):
    j = 0
    for i in range(0, (len(lst)+(runs-(len(lst))))):
        if i < len(lst):
            print(lst[i])
        else:
            print(lst[j])
            j +=1

测试1

# input:
runs = 2
looper(['A', 'B', 'C'], runs)

# output:
A
B

测试2

#input :
runs = 5
looper(['A', 'B', 'C'], runs)

#output :
A
B
C
A
B

如您所见，如果runs变得足够大，这个简单的函数将中断。除此之外，每次调用这个函数时，我想从上次调用结束的索引开始。因此，从上一个示例开始，下一次调用：

runs = 1
looper(['A', 'B', 'C'], runs)

那么输出将是：

c

由于还没有完全了解发电机的使用，我只能猜测这将是一个有用的方法。但是我发现，如果你想按照帖子Generator Expressions vs. List Comprehension中的顶部答案在列表中循环一次，你应该只使用生成器。据我所知，生成器毕竟没有用处，因为Resetting a generator object in Python似乎有点复杂。谢谢你的建议

Answer 1

结合评论中的建议，您可以获得您想要的确切行为。我不会详细讨论发电机是如何工作的，因为您从有经验的人那里得到了更简洁的答案。我将总结一下，生成器的要点是它们yield为您提供下一个元素，只有在您要求时才提供它。生成器和Python中的其他对象一样，都是对象，因此只要您共享同一个生成器对象，您将始终拥有最后一个元素yield的书签。现在要更好地理解代码：

首先，我们将泛化looper函数：

def looper(iterable, runs):
    i = cycle(iterable)
    for _ in range(runs):
        print(next(i))
    return i

因此，现在它允许您在任何iterable上调用一次函数，调用的runs数量与您想要的数量相同。另一方面，如果要多次调用它，它将返回为重用而创建的生成器

现在让我们测试一些案例：

a = [1, 2, 3]

looper(a, 6)

这将打印：

1
2
3
1
2
3

但更重要的是，这也将：

i = looper(a, 2)
looper(i, 2)
looper(i, 2)

---

注意事项：

• 我们之所以能够做到这一切，是因为^{}正在返回一个生成器对象

• 如果希望在调用之间产生循环效果，则必须将返回的生成器传递给后续调用。例如，如果我们尝试几次looper(a, 2)，我们总会得到：

    1
    2
  

  这是因为每次调用函数时，我们都会再次调用cycle，并从从第一个元素开始的原始列表创建一个新的生成器

• 另一方面，当将生成器传递给函数时，cycle函数将创建一个新的生成器，但从该点开始，前一个生成器停止。这样，如果您继续传递第一个返回的cycle生成器，则可以保留循环效果