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

时间:2013-10-30 15:09:44

标签: python list python-3.x

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

例如

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

15 个答案:

答案 0 :(得分:59)

您可以使用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)

答案 1 :(得分:7)

试试这个。

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

答案 2 :(得分:7)

您可以使用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']]

答案 3 :(得分:5)

您知道第二个列表较短吗?

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

这实际上将为您提供一个元组列表,而不是列表列表。我希望可以。

答案 4 :(得分:2)

一种非常简单的方法是将简短列表相乘,使其更长:

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

zip(my_list, another_list*3))
#[(1, 'Yes'), (2, 'No'), (3, 'Yes'), (5, 'No'), (5, 'Yes'), (9, 'No')]

请注意,由于zip仅到达最短列表的长度,因此不需要仔细计算乘数(乘数的目的是确保最短列表为{{1 }}。也就是说,如果使用my_list而不是100,结果将是相同的。

答案 5 :(得分:2)

您可以在循环递增的循环中使用%模运算符

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

new_list = []
for cur in range(len(my_list)):
    new_list.append([my_list[cur], another_list[cur % 2]])
# [[1, 'Yes'], [2, 'No'], [3, 'Yes'], [5, 'No'], [5, 'Yes'], [9, 'No']]

2可以替换为len(another_list)

答案 6 :(得分:1)

对称,没有条件一个班轮

[*zip(A*(len(B)//len(A) + 1), B*(len(A)//len(B) + 1))]

严格回答'如何压缩两个不同大小的列表?'

需要一个相同大小的列表的补丁是通用的:

[*(zip(A, B) if len(A) == len(B)
         else zip(A*(len(B)//len(A) + 1),
                  B*(len(A)//len(B) + 1)))]

答案 7 :(得分:1)

尝试这样:

my_list=[  1,   2,   3, 5,  5,  9]
another_list=['Yes','No']
if type(len(my_list)/2) == float:
  ml=int(len(my_list)/2)+1
else:
  ml=int(len(my_list)/2)

print([[x,y] for x,y in zip(my_list,another_list*ml)])

本机方式:

  • 尝试计算并四舍五入第一个列表长度的一半,如果它是浮动的,则也加1。
  • 在之前乘以zip()并乘以YesNo列表再乘以之前计算出的数字

答案 8 :(得分:1)

让我们使用np.tilezip

my_list = [1, 2, 3, 5, 5, 9]
another_list = ['Yes', 'No']
list(zip(my_list,np.tile(another_list, len(my_list)//len(another_list) + 1)) )

输出:

[(1, 'Yes'), (2, 'No'), (3, 'Yes'), (5, 'No'), (5, 'Yes'), (9, 'No')]

答案 9 :(得分:1)

我喜欢Henry Yik's answer,执行速度更快,但这是不使用itertools的答案。

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

new_list = []
for i in range(len(my_list)):
    new_list.append([my_list[i], another_list[i % len(another_list)]])

new_list

[[1, 'Yes'], [2, 'No'], [3, 'Yes'], [5, 'No'], [5, 'Yes'], [9, 'No']]

答案 10 :(得分: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)

答案 11 :(得分:0)

对于可以按任意顺序使用任意数量的可能无限迭代的版本:

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

答案 12 :(得分:0)

任意数量的可迭代对象的解决方案,并且您不知道哪个最长(也允许任何空的可迭代对象使用默认值):

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)

答案 13 :(得分:0)

d1=['one','two','three']
d2=[1,2,3,4,5]

邮政编码

zip(d1,d2)
<zip object at 0x05E494B8>

zip列表

list(zip(d1,d2))

邮政编码列表字典

{'one': 1, 'two': 2, 'three': 3}

注意:Python 3.7 +

答案 14 :(得分:-1)

现在有list comprehentions

[(i, B[i % 3 - 1]) for i in A]

或者A的元素不是连续的而不是担心列表长度

[(j, B[i % len(B)]) for i, j in enumerate(A)] if len(A) >= len(B) else \
[(A[i % len(A)], j) for i, j in enumerate(B)]