分割陣列依賴於數組值在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.