将两个排序后的数组组合成一个长度相同的新数组

Question

我有两个固定长度 N 的数组。然后我想使用相同的阈值随机切割这两个数组。然后我想将第一个数组的第一部分粘合到第二个数组的第二部分。这个新数组的大小仍应与原始数组相同。

如果我的解释不清楚，请考虑以下示例：

我们有以下两个数组

arr1 = [1,3,5,7,8,10]
arr2 = [2,4,5,7,8,11]

如果我选择 4 作为阈值，这将给出以下新数组

result = [1,3,4,5,7,8,11]

这显然与原始数组的大小不同。但是，如果我们选择 6 作为阈值，我们会得到

result = [1,3,5,7,8,11]

具有所需大小。由于我将经常在相当大的数组上进行此计算，因此我想知道最有效的方法。我正在用 python 写这个，因此更喜欢使用 numpy 方法的答案。

Answer 1

这是一个想法：

# find the first index that arr1[idx] >= arr2[idx]
idx = np.argmax(np.array(arr1) >= arr2)

# for every i < idx <= j
# arr1[i] < arr2[i] < arr2[j]
# so the output is increasing
np.concatenate([arr1[:idx], arr2[idx:]])

输出：

array([ 1,  3,  5,  7,  8, 11])

如果你对阈值感兴趣，那就是arr2[idx]，注意idx == len(arr1) 的情况。