def myzip(x, y):
x, y = iter(x), iter(y)
try:
while True:
yield next(x), next(y)
except StopIteration:
returnSo I tried it with the builtin `zip(a,b)`, and indeed, if a is shorter, `zip(a,b)` consumes the same number of elements from both a and b. But if a is longer, then `zip(a,b)` consumes one extra element from a. I don't know if this behavior is well defined, that is whether I can rely on this, or the order of evaluation of the arguments of a function is up to the interpreter/compiler, and may change.
masklinn wrote a comment in another thread about this that i didn't understand until I saw your implementation.
def myzip(*seqs):
next_yield = [None] * len(seqs)
its = [iter(seq) for seq in seqs]
should_go = True
while should_go:
for index, it in enumerate(its):
try:
next_yield[index] = next(it)
except StopIteration:
should_go = False
if should_go:
yield tuple(next_yield)