如何压缩两个不同大小的列表？

Question

我想压缩两个不同长度的列表

例如

A = [1,2,3,4,5,6,7,8,9]
B = ["A","B","C"]

而且我期待这一点

[(1, 'A'), (2, 'B'), (3, 'C'), (4, 'A'), (5, 'B'), (6, 'C'), (7, 'A'), (8, 'B'), (9, 'C')]

但是内置zip不会重复与更大尺寸的列表配对.是否存在任何内置方式可以实现这一目标？谢谢

这是我的代码

idx = 0
zip_list = []
for value in larger:
    zip_list.append((value,smaller[idx]))
    idx += 1
    if idx == len(smaller):
        idx = 0

Answer 1

你可以使用itertools.cycle:

使迭代器返回迭代中的元素并保存每个元素的副本.当iterable耗尽时,返回保存副本中的元素.无限期地重复.

例:

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)

Answer 2

针对任意数量的可迭代对象的解决方案，并且您不知道哪一个最长（也允许任何空可迭代对象的默认值）：

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)

输出：

(0, 0, 'a', None)
(1, 1, 'b', None)
(0, 2, 'c', None)
(1, 3, 'a', None)
(0, 4, 'b', None)

Answer 3

试试这个.

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)]))

只需确保更大的列表在A.

Answer 4

您可以使用itertools.cycle：

from itertools import cycle

my_list = [1, 2, 3, 5, 5, 9]
another_list = ['Yes', 'No']

cyc = cycle(another_list)

print([[i, next(cyc)] for i in my_list])
# [[1, 'Yes'], [2, 'No'], [3, 'Yes'], [5, 'No'], [5, 'Yes'], [9, 'No']]

Answer 5

你知道第二个名单更短吗？

import itertools
list(zip(my_list, itertools.cycle(another_list)))

这实际上会给你一个元组列表而不是一个列表列表。我希望没关系。