我有一个列表/数组(可以把它们变成数组,没关系):
array = [1,2,1,1,2,3,1,2,1,2,3,1,2,3,4,5]
我想创建一个 new_array来自 array看起来像这样:
new_array = [[1,2],[1],[1,2,3],[1,2],[1,2,3],[1,2,3,4,5]]
我想出的逻辑是从 a=0,b=1 开始并循环遍历 array , 当值 array[a] < array[b] , 然后 a+=1 , b+=1 , 但如果 array[a]>=array[b] 的值, 然后 append b 的值: 这是我尝试过的:
index_1 = [] # This will be the list which will have the FIRST index
index_2 = [] # This will be the list which will have the SECOND index
a = 0
b = 1
for i in array:
if i[a] < i[b]:
a = a+1
b = b+1
elif i[a] >= i[b]:
index_2.append(b)
index_1.append(a)
所以 index_1将有第一个索引和 index_2将有第二个索引,然后我们可以创建 new_array通过:
new_array = [(array[start:end]) for start,end in zip(index_1,index_2)]
不幸的是,即使我的想法是正确的,循环也停止在if i[a]<i[b]。因为IndexError: invalid index to scalar variable. .
请您参考如下方法:
我们可以通过同时形成结果的元素(更准确地说是通过一对相邻元素)直接遍历数组。
我们用数组第一个元素的子列表初始化我们的结果,然后:
array = [1,2,1,1,2,3,1,2,1,2,3,1,2,3,4,5]
new_array = [[array[0]]] # assume the array is not empty
for a,b in zip(array, array[1:]):
if a<b:
new_array[-1].append(b)
else:
new_array.append([b])
print(new_array)
# [[1, 2], [1], [1, 2, 3], [1, 2], [1, 2, 3], [1, 2, 3, 4, 5]]
更新。
或者,我们可以用索引来实现它。
(基于 this answer by @fortran)
首先,准备索引来拆分数组:
idxs_to_split = [idx for idx in range(1, len(array)) if not array[idx-1] < array[idx]]
pairs = zip([0] + idxs_to_split, idxs_to_split + [None])
其次,自己实现拆分:
new_array = [array[i:j] for i,j in pairs]
print(new_array)
# [[1, 2], [1], [1, 2, 3], [1, 2], [1, 2, 3], [1, 2, 3, 4, 5]]
