根据分隔符拆分列表

Question

我有一个包含各种字符串值的列表。每当我看到WORD 时，我都想拆分列表。结果将是一个列表列表（将是原始列表的子列表），其中包含 WORD 的一个实例实现这个？

示例 = ['A', 'WORD', 'B' , 'C' , 'WORD' , 'D']

结果 = [['A'], ['WORD','B','C'],['WORD','D']]

这是我尝试过的，但实际上并没有达到我想要的效果，因为它会将WORD 放在它应该在的不同列表中：

def split_excel_cells(delimiter, cell_data):

    result = []

    temp = []

    for cell in cell_data:
        if cell == delimiter:
            temp.append(cell)
            result.append(temp)
            temp = []
        else:
            temp.append(cell)

    return result

Answer 1

import itertools

lst = ['A', 'WORD', 'B' , 'C' , 'WORD' , 'D']
w = 'WORD'

spl = [list(y) for x, y in itertools.groupby(lst, lambda z: z == w) if not x]

这会创建一个没有分隔符的拆分列表，这对我来说看起来更合乎逻辑：

[['A'], ['B', 'C'], ['D']]

如果您坚持要包含分隔符，这应该可以解决问题：

spl = [[]]
for x, y in itertools.groupby(lst, lambda z: z == w):
    if x: spl.append([])
    spl[-1].extend(y)

Answer 2

我会使用生成器：

def group(seq, sep):
    g = []
    for el in seq:
        if el == sep:
            yield g
            g = []
        g.append(el)
    yield g

ex = ['A', 'WORD', 'B' , 'C' , 'WORD' , 'D']
result = list(group(ex, 'WORD'))
print(result)

打印出来

[['A'], ['WORD', 'B', 'C'], ['WORD', 'D']]

代码接受任何可迭代对象，并生成一个可迭代对象（如果您不想这样做，则必须将其展平为列表）。

Answer 3

给定

import more_itertools as mit


iterable = ["A", "WORD", "B" , "C" , "WORD" , "D"]
pred = lambda x: x == "WORD"

代码

list(mit.split_before(iterable, pred))
# [['A'], ['WORD', 'B', 'C'], ['WORD', 'D']]

more_itertools 是可通过> pip install more_itertools 安装的第三方库。

另请参阅 split_at 和 split_after。

Answer 4

• @NPE 的解决方案在我看来非常 Pythonic。这是另一个使用itertools：
• izip 特定于 python 2.7。将 izip 替换为 zip 以在 python 3 中工作

from itertools import izip, chain
example = ['A', 'WORD', 'B' , 'C' , 'WORD' , 'D']
indices = [i for i,x in enumerate(example) if x=="WORD"]
pairs = izip(chain([0], indices), chain(indices, [None]))
result = [example[i:j] for i, j in pairs]

• 这段代码主要基于this answer。