我编写了一个函数,它有两个参数:列表和先前给出的列表中存在的一个值(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
答案 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)