使用分隔符拆分列表

时间:2014-10-28 13:49:00

标签: python

我编写了一个函数,它有两个参数:列表先前给出的列表中存在的一个值(sep)。该函数的目的是拆分给定列表并返回列表中的多个列表,而不包含在写入函数的第二个参数中指定的值。所以使用def split_list([1,2,3,2,1],2)--->结果将是[[1],[3],[1]]。拆分功能很好,但结果会将函数(sep)的第二个值保留在单独的列表中。我想不出如何解决这个问题的方法。提前致谢

def split_list(l, sep):
 occurence = [i for i, x in enumerate(l) if x == sep]
 newlist=[]
 newlist.append(l[:occurence[0]])
 for i in range(0,len(occurence)):
  j=i+1

  if j < len(occurence):
   newlist.append(l[occurence[i]:occurence[j]])
  i+=1
 newlist.append(l[occurence[-1]:])


 return newlist 

3 个答案:

答案 0 :(得分:2)

这个怎么样:

def split_list(l, sep):
    nl = [[]]
    for el in l:
        if el == sep:
            nl.append([])
        else:
            # Append to last list
            nl[-1].append(el)
    return nl

或者使用您的方法,使用出现的列表:

def split_list(l, sep):
    # occurences
    o = [i for i, x in enumerate(l) if x == sep]
    nl = []
    # first slice
    nl.append(l[:o[0]])
    # middle slices
    for i in range(1, len(o)):
        nl.append(l[o[i-1]+1:o[i]])
    # last slice
    nl.append(l[o[-1]+1:])
    return nl

答案 1 :(得分:2)

您可以使用以下列表理解和zip功能分割您的列表:

>>> l=[1,2,3,2,1,8,9]
>>> oc= [i for i, x in enumerate(l) if x == 2]
>>> [l[i:j] if 2 not in l[i:j] else l[i+1:j] for i, j in zip([0]+oc, oc+[None])]
[[1], [3], [1, 8, 9]]

所以对你的功能:

def split_list(l, sep):
 occurence = [i for i, x in enumerate(l) if x == sep]
 return [l[i:j] if sep not in l[i:j] else l[i+1:j] for i, j in zip([0]+occurence, occurence+[None])]

答案 2 :(得分:0)

如果x!= sep则使用[list(x)代表i,x代表枚举(l)