63

長さの異なる2つのリストを圧縮したい

例えば

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
4

16 に答える 16

102

使用できますitertools.cycle：

iterable から要素を返すイテレータを作成し、それぞれのコピーを保存します。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)
于 2013-10-30T15:13:36.907 に答える
9

これを試して。

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ください。

于 2013-10-30T15:15:41.817 に答える
6

任意の数の iterable の解決策で、どれが最も長いかわからない (空の iterable のデフォルトも許可する):

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)
于 2019-01-25T22:09:01.677 に答える
1

任意の順序で有限数の潜在的に無限の iterable で動作するバージョンの場合:

from itertools import cycle, tee, zip_longest

def cyclical_zip(*iterables):
    iterables_1, iterables_2 = zip(*map(tee, iterables))  # Allow proper iteration of iterators

    for _, x in zip(
            zip_longest(*iterables_1),      # Limit             by the length of the longest iterable
            zip(*map(cycle, iterables_2))): #       the cycling
        yield x

assert list(cyclical_zip([1, 2, 3], 'abcd', 'xy')) == [(1, 'a', 'x'), (2, 'b', 'y'), (3, 'c', 'x'), (1, 'd', 'y')]  # An example and test case
于 2018-12-22T00:34:34.710 に答える
0

おそらくもっと良い方法がありますが、リストを必要な長さに繰り返す関数を作成できます。

def repeatlist(l,i):
    '''give a list and a total length'''
    while len(l) < i:
        l += l
    while len(l) > i:
        l.pop()

それからする

repeatlist(B,len(A))
zip_list = zip(A,B)
于 2013-10-30T15:22:57.173 に答える