根据另一个数组中的值，从一个数组创建多个 arrays - python

Question

情况如下：

假设我有两个 arrays，x 和 y： 输入：

x = [1,2,3,4,5,6,7,8,9,10,11,12,13,14]
y = [2,6,9,13]

预期 output：

arr1 = [2,3,4,5]
arr2 = [6,7,8]
arr3 = [9,10,11,12]

我想创建一个 python 脚本，它可以让我根据数组 y 的值作为端点将数组 x 拆分为多个 arrays。

所以 x 将在 2 和 6 之间分裂，然后在 6 和 9 之间，然后在这个例子中在 9 和 13 之间。

我不知道如何开始，我是初学者。 我会很感激你的帮助，我很想知道你是如何分解问题来解决它的？ 谢谢！

Answer 1

对于x排序，我们可以使用np.searchsorted和np.split来拆分x使用y包含在x中的索引：

import numpy as np

i = np.searchsorted(x, y)
np.split(x,i+1)[1:-1]
# [array([3, 4, 5, 6]), array([7, 8, 9]), array([10, 11, 12, 13])]

Answer 2

根据 y 中的值找到 x 的索引，然后使用索引

arr = []
for i in range(len(y)-1):
    arr.append(x[x.index(y[i]):x.index(y[i+1])])

arr

[[2, 3, 4, 5], [6, 7, 8], [9, 10, 11, 12]]

如果数组中有重复项，则此方法有效，

x = [2,3,4,5,6,7,7,8,9,9,10,11,11,12,13,14,15]
y = [2,6,9,13]

arr = []
for i in range(len(y)-1):
    arr.append(x[x.index(y[i]):x.index(y[i+1])])

[[2, 3, 4, 5], [6, 7, 7, 8], [9, 9, 10, 11, 11, 12]]

Answer 3

您的预期 output 有点不一致，或者我没有得到它背后的模式，但这应该让您接近您想要的

x = [1,2,3,4,5,6,7,8,9,10,11,12,13,14]
y = [2,6,9,13]

m = []

for i in range(len(y) - 1):
    m += [x[y[i] - 1:y[i+1] - 1]]

print(m)

Output

[[2, 3, 4, 5], [6, 7, 8], [9, 10, 11, 12]]