我有一个整数数组
[1,2,3]
,需要返回所有可能的连续子数组组合。
[[1],[2],[3],[1,2],[2,3],[1,2,3]]
如何使用Python处理呢?一种方法是使用两个循环和数组本身,但应该有更好的方法。
[1,2,3]
,需要返回所有可能的连续子数组组合。
[[1],[2],[3],[1,2],[2,3],[1,2,3]]
如何使用Python处理呢?一种方法是使用两个循环和数组本身,但应该有更好的方法。
一行解决方案(我不知道"更好的方式"对你来说是什么意思)
L = [1,2,3]
[L[i:i+j] for i in range(0,len(L)) for j in range(1,len(L)-i+1)]
L=[1,2,3,4]
[L[i:i+j] for i in range(0,len(L)) for j in range(1,len(L)-i+1)]
你得到了,
[[1], [1, 2], [1, 2, 3], [2], [2, 3], [3]]
[[1],
[1, 2],
[1, 2, 3],
[1, 2, 3, 4],
[2],
[2, 3],
[2, 3, 4],
[3],
[3, 4],
[4]]
简化检查员的解决方案:
def getAllWindows(L):
for w in range(1, len(L)+1):
for i in range(len(L)-w+1):
yield L[i:i+w]
并且有一种不使用循环的解决方案:
def allSubArrays(L,L2=None):
if L2==None:
L2 = L[:-1]
if L==[]:
if L2==[]:
return []
return allSubArrays(L2,L2[:-1])
return [L]+allSubArrays(L[1:],L2)
def kwindow(L, k):
for i in range(len(L)-k+1):
yield L[i:i+k]
def getAllWindows(L):
for w in range(1, len(L)+1):
yield from kwindow(L, w)
输出:
In [39]: for i in getAllWindows([1,2,3]): print(i)
[1]
[2]
[3]
[1, 2]
[2, 3]
[1, 2, 3]
itertools
的方法:import itertools
def allSubArrays(xs):
n = len(xs)
indices = list(range(n+1))
for i,j in itertools.combinations(indices,2):
yield xs[i:j]
例如:
>>> list(allSubArrays([1,2,3]))
[[1], [1, 2], [1, 2, 3], [2], [2, 3], [3]]
li=[1,2,3]
l=[]
for i in range(length(li)):
for j in range(i,len(li)+1):
if i==j: *cancelling empty sublist item*
continue
else:
subli=li[i:j]
l.append(subli)
print(l)
输出:
[[1], [1, 2], [1, 2, 3], [2], [2, 3], [3]]