python - Numpy:获取索引大于值且条件为真的数组

Question

我有以下数组:

a = np.array([6,5,4,3,4,5,6])

现在我想获取所有大于 4 但索引值大于 2 的元素。我发现这样做的方法如下:

a[2:][a[2:]>4]

是否有更好或更具可读性的方法来完成此任务？

更新:这是一个简化版本。实际上，索引是通过对几个变量的算术运算完成的，如下所示:

a[len(trainPredict)+(look_back*2)+1:][a[len(trainPredict)+(look_back*2)+1:]>4]

trainPredict 是一个 numpy 数组，look_back 是一个整数。
我想看看是否有既定的方法或其他人如何做到这一点。

Answer 1

如果您担心切片的复杂性和/或条件的数量，您总是可以将它们分开:

a = np.array([6,5,4,3,4,5,6])

a_slice = a[2:]

cond_1 = a_slice > 4

res = a_slice[cond_1]

您的示例是否非常简化？对于更复杂的操作，可能有更好的解决方案。