blob: 8f47679aaf75c79d6588d810bd4b60790c5d43da (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
|
import itertools
import random
import typing as t
class RandomCycle:
"""
Cycles through elements from a randomly shuffled iterable, repeating indefinitely.
The iterable is reshuffled after each full cycle.
"""
def __init__(self, iterable: t.Iterable) -> None:
self.iterable = list(iterable)
self.index = itertools.cycle(range(len(iterable)))
def __next__(self) -> t.Any:
idx = next(self.index)
if idx == 0:
random.shuffle(self.iterable)
return self.iterable[idx]
|
