np.argwhere给出了切片中的索引。我们已经知道如何将切片中的索引转换为数组中的索引：切片中的索引x对应于数组中的索引i[x]，所以我们只需要用np.argwhere的结果索引到i！

A = np.array([1,2,3,4,5,6,7,8,9])
i = np.array([0, 3, 6, 7, 8])

slice_lt8 = np.argwhere(A[i] < 8)
arr_lt8 = i[slice_lt8]

arr_lt8现在是期望值：

array([[0],
       [3],
       [6]])

import numpy as np

A = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9])
i = [0, 3, 6, 7, 8]
argA = np.argwhere(A[i]<8)
argList = [argA[idx][0] for idx in range(len(argA))]

desList = [i[idx] for idx in argList]
desListOfLists = [[x] for x in desList]
desArray = np.array(desListOfLists)

由于其他读者可能想要稍微不同的输出类型，我已经包括了三个：list; list of lists和ndarray：
desList生成

[0, 3, 6]

desListOfLists生成

[[0], [3], [6]]

desArray生成

array([[0], [3], [6]])

这是最初的要求