在 Python 中分割列表列表

Question

我有一个长度相同的列表列表。我想将第一个列表分割成给定值的连续运行。然后我想分割剩余的列表以匹配从第一个列表生成的段。例如：Given value: 2Given list of lists: [[0,0,2,2,2,1,1,1,2,3], [1,2,3,4,5,6,7,8,9,10], [1,1,1,1,1,1,1,1,1,1]Return: [ [[2,2,2],[2]], [[3,4,5],[9]], [[1,1,1],[1]] ]我得到的最接近的是通过以下方式获取索引：>>> import itertools>>> import operator>>> x = 2>>> L = [[0,0,2,2,2,1,1,1,2,3],[1,2,3,4,5,6,7,8,9,10],[1,1,1,1,1,1,1,1,1,1]]>>> I = [[i for i,value in it] for key,it in itertools.groupby(enumerate(L[0]), key=operator.itemgetter(1)) if key == x]>>> print I[[2, 3, 4], [8]]此代码是根据此站点上的另一个问题修改的。我想找到最有效的方法，因为这些列表可能很长。编辑：也许如果我将列表放在一起，可能会更清楚：[[0,0,[2,2,2],1,1,1,[2],3], -> [2,2,2],[2] [1,2,[3,4,5],6,7,8,[9],10],-> [3,4,5],[9] [1,1,[1,1,1],1,1,1,[1],1]] -> [1,1,1],[1]

Answer 1

您可以使用groupby以组的起始索引和长度的元组的形式创建组列表，并使用此列表从每个子列表中提取值：

from itertools import groupby

from operator import itemgetter

def match(L, x):

    groups = [(next(g)[0], sum(1 for _ in g) + 1)

        for k, g in groupby(enumerate(L[0]), key=itemgetter(1)) if k == x]

    return [[lst[i: i + length] for i, length in groups] for lst in L]

以便：

match([[0,0,2,2,2,1,1,1,2,3], [1,2,3,4,5,6,7,8,9,10], [1,1,1,1,1,1,1,1,1,1]], 2)

返回：

[[[2, 2, 2], [2]], [[3, 4, 5], [9]], [[1, 1, 1], [1]]]

Answer 2

l=[[0,0,2,2,2,1,1,1,2,3], [1,2,3,4,5,6,7,8,9,10], [1,1,1,1,1,1,1,1,1,1]]

temp=l[0]

value=2

dict={}

k=-1

prev=-999

for i in range(0,len(temp)):

    if(temp[i]==value):

        if(prev!=-999 and prev==i-1):

            if(k in dict):

                dict[k].append(i)

            else:

                dict[k]=[i]

        else:

            k+=1

            if(k in dict):

                dict[k].append(i)

            else:

                dict[k]=[i]

        prev=i

output=[]

for i in range(0,len(l)):

    single=l[i]

    final=[]

    for keys in dict: #{0: [2, 3, 4], 1: [8]}

        ans=[]

        desired_indices=dict[keys]

        for j in range(0,len(desired_indices)):

            ans.append(single[desired_indices[j]])

        final.append(ans)

    output.append(final)

print(output) #[[[2, 2, 2], [2]], [[3, 4, 5], [9]], [[1, 1, 1], [1]]]

这似乎是一种方法，首先创建传染性元素的字典，然后在每个列表中查找该键并存储在输出中。