分割阵列依赖于数组值在Python

Question

我有坐标的阵列是这样的：

array = [[1,6],[2,6],[3,8],[4,10],[5,6],[5,7],[18,6],[19,5],[17,9],[10,5]] 

欲6之间的阵列分开。和7。坐标，因为那里的值为X。我想获得两个独立阵列，分别为arr1和arr2，其中arr1是拆分前的值，而arr2是后面的值。

我想说的是，如果下一个X值大于10相差较大，将追加到arr2，否则arr1，这样的事情：

arr1 = [] 
arr2 = [] 
for [x,y] in array: 
    if next(x) > 10: 
     arr2.append(x,y) 
    else: 
     arr1.append(x,y) 

是否有人可以帮我解决这个问题？

Answer 1

你可以做到以下几点：

ar = np.array([[1,6],[2,6],[3,8],[4,10],[5,6],[5,7],[18,6],[19,5],[17,9],[10,5]]) 

# get differences of x values 
dif = ar[1:, 0] - ar[:-1, 0] 

# get the index where you first observe a jump 
fi = np.where(abs(dif) > 10)[0][0] 

ar1 = ar[:fi+1] 
ar2 = ar[fi+1:] 

然后dif是：

array([ 1, 1, 1, 1, 0, 13, 1, -2, -7]) 

fi是5和ar1和ar2是：

array([[ 1, 6], 
     [ 2, 6], 
     [ 3, 8], 
     [ 4, 10], 
     [ 5, 6], 
     [ 5, 7]]) 

和

array([[18, 6], 
     [19, 5], 
     [17, 9], 
     [10, 5]]), 

。

这也将让你得到你的数据的所有跳跃（你只需要改变fi = np.where(abs(dif) > 10)[0][0]到fi = np.where(abs(dif) > 10)[0]）

Answer 2

试试这个：

prev = array[0][0]
pos = -1
for i in range (1, len(array)):
if array[i][0] - prev >1:
break
else:
prev = array[i][0]
if pos != -1:
arr1 = array[:pos]
arr2 = array[pos:]

这应该拆分array你想要的方式。需要注意的是列表是从0

Answer 3

索引这可能是你在找什么

array = [[1,6],[2,6],[3,8],[4,10],[5,6],[5,7],[18,6],[19,5],[17,9],[10,5]] 
# Declare two array variables 
arr1 = None 
arr2 = None 
n = len(array) 
for i in range(n-1): 
    if abs(array[i][0] - array[i+1][0]) >= 10: 
     arr1 = array[:i+1] 
     arr2 = array[i+1:] 
     break 

print arr1 
print arr2 

Answer 4

arry1 = [] 
arry2 = [] 
for i in arry: 
    if (i[0] - i[1]) > 10: 
     arry1.append(i) 
    else: 
     arry2.append(i) 

Answer 5

如果我得到正确你的问题，你想数组在第一点里劈随后的两个x值之间的差值大于10 你可以做，使用numpy的：

import numpy as np 
THRESH = 10 
array = [[1,6],[2,6],[3,8],[4,10],[5,6],[5,7],[18,6],[19,5],[17,9],[10,5]] 
array = np.asarray(array) 
deltas_x = np.abs(array[1:,0] - array[:-1,0]) 
split_idx = np.where(deltas_x > THRESH)[0][0] + 1 
arr1 = array[:split_idx,:] 
arr2 = array[split_idx:,:] 

需要注意的是，我们需要加1 np.where的结果占发克拉该deltas_x阵列是1个值大于array

Answer 6

可能最容易在连续对与分割只要间隙被发现迭代较短：

array = [[1,6],[2,6],[3,8],[4,10],[5,6],[5,7],[18,6],[19,5],[17,9],[10,5]] 

for idx, (cur, nxt) in enumerate(zip(array, array[1:])): # successive pairs and index 
    if abs(cur[0] - nxt[0]) > 10: # compare difference of first items 
     arr1, arr2 = array[:idx+1], array[idx+1:] # split 
     break # no further splits, end the loop now 
else: # no break, keep the original array 
    arr1, arr2 = array, [] 

其中给出：

>>> arr1 
[[1, 6], [2, 6], [3, 8], [4, 10], [5, 6], [5, 7]] 
>>> arr2 
[[18, 6], [19, 5], [17, 9], [10, 5]] 

它如果你想分几次就会有点困难，但这应该适用于你的情况。

Answer 7

当比较连续元素我通常使用枚举：

array = [[1,6],[2,6],[3,8],[4,10],[5,6],[5,7],[18,6],[19,5],[17,9],[10,5]] 
arr1 = list() 
arr2 = list() 
gap = 10 

for index, value in enumerate(array[:-1]): # [:-1] prevents out of range 
    if abs(value[0]-array[index+1][0]) >= gap: 
     arr1.append(value) 
    else: 
     arr2.append(value) 

arr2.append(array[-1]) # Take into account that the last element needs to be added to one of the arrays.