¿Cómo comprimir dos listas de diferentes tamaños, repitiendo la lista más corta?
Quiero comprimir dos listas con longitud diferente
Por ejemplo
A = [1,2,3,4,5,6,7,8,9]
B = ["A","B","C"]
y espero esto
[(1, 'A'), (2, 'B'), (3, 'C'), (4, 'A'), (5, 'B'), (6, 'C'), (7, 'A'), (8, 'B'), (9, 'C')]
Pero el integrado zipno se repetirá para emparejarse con la lista de mayor tamaño. ¿Existe alguna forma integrada de lograr esto?
Aquí está mi código:
idx = 0
zip_list = []
for value in larger:
zip_list.append((value,smaller[idx]))
idx += 1
if idx == len(smaller):
idx = 0
Aceptado
Puedes usar itertools.cycle:
Haga un iterador que devuelva elementos del iterable y guarde una copia de cada uno. Cuando se agote el iterable, devuelva elementos de la copia guardada. Se repite indefinidamente.
Ejemplo:
A = [1,2,3,4,5,6,7,8,9]
B = ["A","B","C"]
from itertools import cycle
zip_list = zip(A, cycle(B)) if len(A) > len(B) else zip(cycle(A), B)
Solución para un número arbitrario de iterables y no sabe cuál es el más largo (también permite un valor predeterminado para cualquier iterable vacío):
from itertools import cycle, zip_longest
def zip_cycle(*iterables, empty_default=None):
cycles = [cycle(i) for i in iterables]
for _ in zip_longest(*iterables):
yield tuple(next(i, empty_default) for i in cycles)
for i in zip_cycle(range(2), range(5), ['a', 'b', 'c'], []):
print(i)
Salidas:
(0, 0, 'a', None)
(1, 1, 'b', None)
(0, 2, 'c', None)
(1, 3, 'a', None)
(0, 4, 'b', None)
Prueba esto.
A = [1,2,3,4,5,6,7,8,9]
B = ["A","B","C"]
Z = []
for i, a in enumerate(A):
Z.append((a, B[i % len(B)]))
Solo asegúrese de que la lista más grande esté en formato A.
