¿Se puede hacer más simple/elegante?Programa de Python para dividir una lista en dos listas con elementos alternados
def zigzag(seq):
"""Return two sequences with alternating elements from `seq`"""
x, y = [], []
p, q = x, y
for e in seq:
p.append(e)
p, q = q, p
return x, y
Si seq, como usted dice, es una lista, a continuación:
def zigzag(seq):
return seq[::2], seq[1::2]
Si seq es un iterable totalmente genérico, como posiblemente un generador:
def zigzag(seq):
results = [], []
for i, e in enumerate(seq):
results[i%2].append(e)
return results
def zigzag(seq):
return seq[::2], seq[1::2]
Solo funciona para las listas, mientras que la otra solución funciona para cualquier iterable. –
cierto, pero el título especificó una lista – cobbal
Esto toma un iterador y devuelve dos iteradores:
import itertools
def zigzag(seq):
t1,t2 = itertools.tee(seq)
even = itertools.islice(t1,0,None,2)
odd = itertools.islice(t2,1,None,2)
return even,odd
Si prefiere listas, entonces puede return list(even),list(odd).
SO está haciendo tan vago. –
@Sridhar, no pienses que es flojo, piensa que es * eficiente en el tiempo *. Pasé media hora trabajando en un algoritmo para hacer esto con un bucle * for * (para * x * columnas, no solo dos). Y aunque lo conseguí funcionando, simplemente no parecía * pitónico * - Sospeché que había una manera más fácil. Efectivamente, no recordaba que las listas tengan * variables * de paso * (como se muestra en esta Respuesta), lo que lo hace trivial. –
Esto es hermoso, ni siquiera sabía acerca de esa sintaxis. ¡Gracias! – sleepycal