如何交错列表

2024-10-01 15:30:25 发布

您现在位置:Python中文网/ 问答频道 /正文

我有两个长度不相等的列表,我想把它们交错起来。我希望能够将额外值附加到交错列表末尾的较长列表中。我有这样一个:

def interleave(xs,ys):
a=xs
b=ys
minlength=[len(a),len(b)]
extralist= list()
interleave= list()
for i in range((minval(minlength))):
    pair=a[i],b[i]
    interleave.append(pair)
    flat=flatten(interleave)
    c=a+b
if len(b)>len(a):
    remainder=len(b)-len(a)
    print(remainder)
    for j in range(remainder,-1,-1):
        extra=b[-j]
        extralist.append(extra)
        print(extralist)
if len(a)>len(b):
    remainder=len(a)-len(b)
    print(remainder)
    for j in range(remainder,-1,-1):
        extra=a[-j]
        extralist.append(extra)
final=flat+extralist
return final

但如果我测试一下:

^{pr2}$

假的和33号不会出现。我做错什么了?在

编辑:我找到了解决办法。我向后索引:)


Tags: in列表forlenrangeextralistprint
2条回答

另一个选择

from itertools import chain
x = [1, 2, 3]
y = iter(["hi", "bye",True, False, 33])
list(chain.from_iterable(zip(x, y))) + list(y)
[1, 'hi', 2, 'bye', 3, True, False, 33]

下面是一个使用^{}list comprehension的解决方案:

>>> from itertools import zip_longest
>>> a = [1, 2, 3]
>>> b = ["hi", "bye", True, False, 33]
>>> [y for x in zip_longest(a, b) for y in x]
[1, 'hi', 2, 'bye', 3, True, None, False, None, 33]
>>> [y for x in zip_longest(a, b) for y in x if y is not None]
[1, 'hi', 2, 'bye', 3, True, False, 33]
>>>

编辑:

如果您不想使用itertools,我想您可以:

^{pr2}$

但是我必须说这个方法没有我的第一个解决方案有效。在


编辑2:

enumerate改为{}又会降低效率。但是,如果您必须执行以下操作:

>>> a = [1, 2, 3]
>>> b = ["hi", "bye", True, False, 33]
>>> out = []
>>> items = sorted((a,b), key=len)
>>> for x,y in zip(range(len(items[1])), items[1]):
...     out.append(y)
...     try:
...         out.append(items[0][x])
...     except IndexError:
...         out.extend(items[1][x+1:])
...         break
...
>>> out
['hi', 1, 'bye', 2, True, 3, False, 33]
>>>

相关问题 更多 >

    热门问题