根据 Python 中的数组值拆分数组

我有一个像这样的坐标数组：

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

我想将数组拆分为6. and 7。协调([5,7],[18,6])因为有一个间隙X那里的价值。我想得到两个单独的数组，arr1 and arr2, where arr1是分割前的值，arr2是后面的值。

我想说的是，如果接下来X值大于差异10，它将附加到arr2, else arr1，像这样：

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

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

如果我正确地回答了你的问题，那么你正尝试在第一个点拆分数组，其中两个后续 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.where 的结果加 1 来说明deltas_x数组比 1 个值短array