将阵列拆分取决于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。协调([5,7],[18,6])，因为那里的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

如果我正确地得到了您的问题，您正在尝试在第一个点分开数组，其中两个随后的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:,:]

请注意，我们需要在NP的结果中添加1个，以说明deltas_x数组比array

Answer 2

这可能是您正在寻找的

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 3

您可以执行以下操作：

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 4

可能最容易在连续的对上迭代并在发现差距后立即拆分：

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 5

在比较连续元素时，我通常会枚举：

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.

Answer 6

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

Answer 7

尝试以下：

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。