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

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]

你的例子很简单吗？对于更复杂的操作，可能有更好的解决方案。你知道吗