Python - 如何连接列表中由特定值分隔的元素

时间:2015-06-05 20:07:58

标签: python list iteration

假设我有一个如下所示的列表:

['1', '+', '2', '+', '3', 'C', '4', 'C', '5', '-', '6', '+', '7', 'C', '8']

我想以这样的方式修改它,即由'C'分隔的所有元素连接在一起。结果应如下所示:

['1', '+', '2', '+', '345', '-', '6', '+', '78']

我一直在尝试迭代地执行此操作,但问题是每当我删除一个元素时,列表的大小会发生变化并且迭代会发生变化。任何指针都表示赞赏。谢谢!

7 个答案:

答案 0 :(得分:4)

过滤掉" C"然后分组连续数字:

from itertools import groupby

data = ['1', '+', '2', '+', '3', 'C', '4', 'C', '5', '-', '6', '+', '7', 'C', '8']
new_data = [''.join(g) if k else next(g) for k, g in groupby((el for el in data if el != 'C'), key=str.isdigit)]
# ['1', '+', '2', '+', '345', '-', '6', '+', '78']

答案 1 :(得分:3)

如果下一个值不是C,您可以使用仅生成值的生成器函数,否则累积:

def join_by(it, sep):
    it = iter(it)
    prev = next(it)
    for elem in it:
        if elem == sep:
            prev += next(it)
        else:
            yield prev
            prev = elem
    yield prev

演示;这不仅限于数字,也不限于特定的分隔符:

>>> orig = ['1', '+', '2', '+', '3', 'C', '4', 'C', '5', '-', '6', '+', '7', 'C', '8']
>>> list(join_by(orig, 'C'))
['1', '+', '2', '+', '345', '-', '6', '+', '78']
>>> list(join_by(['foo', '-', 'bar', 'baz'], '-'))
['foobar', 'baz']

要解决这个难题,你可以在这里作弊并使用eval()

>>> import operator
>>> from itertools import product
>>> digits = '123456789'
>>> ops = (' - ', ' + ', '')
>>> for combo in product(ops, repeat=len(digits) - 1):
...     expr = zip(digits, combo + ('',))
...     expr = ''.join([c for digit_oper in expr for c in digit_oper])
...     result = eval(expr)
...     if result == 100:
...         print(expr)
... 
1 + 2 + 3 - 4 + 5 + 6 + 78 + 9
1 + 2 + 34 - 5 + 67 - 8 + 9
1 + 23 - 4 + 5 + 6 + 78 - 9
1 + 23 - 4 + 56 + 7 + 8 + 9
12 - 3 - 4 + 5 - 6 + 7 + 89
12 + 3 - 4 + 5 + 67 + 8 + 9
12 + 3 + 4 + 5 - 6 - 7 + 89
123 - 4 - 5 - 6 - 7 + 8 - 9
123 - 45 - 67 + 89
123 + 4 - 5 + 67 - 89
123 + 45 - 67 + 8 - 9

答案 2 :(得分:0)

lst = ['1', '+', '2', '+', '3', 'C', '4', 'C', '5', '-', '6', '+', '7', 'C', '8']
newlist = [lst[0]]
for idx, item in enumerate(lst[1:]):
    if item == 'C':
        continue # do nothing if it's a C
    if lst[idx] == 'C':
        newlist[-1] += item # add it to the last item if it follows a C
    else:
        newlist.append(item) # add it to the list

结果:

['1', '+', '2', '+', '345', '-', '6', '+', '78']

答案 3 :(得分:0)

此代码适用于任何给定的列表和值。

old_list = ['1', '+', '2', '+', '3', 'C', '4', 'C', '5', '-', '6', '+', '7', 'C', '8']
new_list = []
for index, val in enumerate(old_list):
    if val == 'C':
        new_list[-1] += old_list[index+1]
    elif old_list[index-1] != 'C':
        new_list.append(val)

答案 4 :(得分:0)

不需要列表理解,只需找到你的分隔符并连接其余部分

a = ['1', '+', '2', '+', '3', 'C', '4', 'C', '5', '-', '6', '+', '7', 'C', '8']
firstIndex = a.index('C')
revertedLastIndex = a[::-1].index('C') # index in reverted list 
lastIndex = len(a) - 1 - revertedLastIndex  # -1 for 0-indexed lists

因此,我们现在可以加入firstIndexlastIndex之间的列表部分,并将其余部分连接起来

newList = a[:firstIndex] + ["".join(a[firstIndex:lastIndex])] + a[lastIndex:]

编辑:“不需要列表理解”我的意思当然不需要显式迭代,index函数在内部执行

答案 5 :(得分:0)

import ast

L=['1', '+', '2', '+', '3', 'C', '4', 'C', '5', '-', '6', '+', '7', 'C', '8']

s=str(L).replace(", 'C', ", "")

out=ast.literal_eval(s)

print out

输出:

['1', '+', '2', '+', '345', '-', '6', '+', '78']

答案 6 :(得分:0)

Python的花哨功能非常好,因为它们使代码的逻辑非常容易理解。如果他们不这样做,我宁愿保持简单和可验证:

stuff = ['1', '+', '2', '+', '3', 'C', '4', 'C', '5', '-', '6', '+', '7', 'C', '8']
concat = False
newstuff = []
for a in stuff:
    if concat:
        newstuff[-1] += a
        concat = False
    elif a == "C":
        concat = True
    else:
        newstuff.append(a)