Question

说我有两个清单

[16, 0, 0, ';', 17, 0, 2, ';', 0, 2, 1, ';']
  [-1, 0, ';', 0, -2, ';', -2, -1, ';']

是否有更简单的方法在满足';'时有条件地合并这些列表元素字符而不是单独迭代它们并将它们组合起来？

输出应为

[16, 0, 0, -1, 0, ';', 17, 0, 2, 0, -2, ';', 0, 2, 1, -2, -2,';']

Answer 1

def get_part(lst,sep=';'):
    out = []
    for i in lst:
        if i == sep:
           yield out
           out = []
        else:
           out.append(i)

现在我们可以将您的列表压缩在一起：

merged = []
for l1,l2 in zip(get_part(list1),get_part(list2)):
    merged.extend(l1)
    merged.extend(l2)
    merged.append(';')

当然，在它的核心，这基本上是迭代它们并结合......所以在回答你的问题时，我认为没有更好的方法。

存储列表的更好方法可能是将它们存储为列表列表 - 例如：

list1 = [[16, 0, 0], [ 17, 0, 2], [ 0, 2, 1] ]

像这样存储，你可以这样做：

merged = [ l1 + l2 for l1,l2 in zip(list1,list2) ]

Answer 2

>>> from itertools import groupby, izip, chain
>>> L1=[16, 0, 0, ';', 17, 0, 2, ';', 0, 2, 1, ';']
>>> L2=[-1, 0, ';', 0, -2, ';', -2, -1, ';']    
>>> g1 = groupby(L1, key=';'.__eq__)
>>> g2 = groupby(L2, key=';'.__eq__)
>>> [i for i1,i2 in izip(g1, g2) for i in ([';'] if i1[0] is True else chain(i1[1], i2[1]))]
[16, 0, 0, -1, 0, ';', 17, 0, 2, 0, -2, ';', 0, 2, 1, -2, -1, ';']

Answer 3

更多itertools疯狂：

from itertools import groupby, chain, izip, repeat

a = [16, 0, 0, ';', 17, 0, 2, ';', 0, 2, 1, ';']
b = [-1, 0, ';', 0, -2, ';', -2, -1, ';']

wanted = [16, 0, 0, -1, 0, ";",
          17, 0, 2, 0, -2, ';',
          0, 2, 1, -2, -1,";"]

def split(items, sep=";"):
    return (group for key, group in
            groupby(items, lambda item: item != sep) if key)

got = list(
    chain.from_iterable(
        chain.from_iterable(
            izip(split(a), split(b), repeat([";"])))))
assert got == wanted

有条件地合并列表

3 个答案: