在另一个数组中查找其对应值应大于M的numpy数组的N个最大索引

Question

我有3 Numpy arrays each of length 107952899。

说：

1. Time = [2.14579526e+08 2.14579626e+08 2.14579726e+08 ...1.10098692e+10  1.10098693e+10]
2. Speed = [0.66 0.66 0.66 .............0.06024864 0.06014756]

3. Brak_press = [0.3, 0.3, 0.3 .............. 0.3, 0.3]

是什么意思

“时间”中的每个索引值对应于“速度和制动”数组中的相同索引值。

Time                Speed         Brake
2.14579526e+08      0.66          0.3
.
.

要求

否1：我想find the indices in Speed array的{​​{1}}里面是values

第2条：对于那些索引，greater than 20

否3：现在我想找到what will be values in Brake Array并将其存储在另一个列表/数组中

所以最后，如果我从Top N Maximum Value indices in Brake Array中提取一个索引并在Brake＆Speed数组中使用它，它必须显示出来。

Top N Maximum Indices

概述

简单来说，我需要的是找到最大N个制动点索引，其对应的速度值应大于20

我尝试过的

Brake[idx] = valid Value & more importantly Speed [idx] = Value > than 20

之后，我尝试了argsort / argpartition，但结果都不符合我的要求

请求

我相信会有最好的方法来做到这一点。。。

（我将上面的 speed_20 = np.where(Speed > 20) # I got indices as tupple brake_values = Brake[speed_20] # Found the Brake Values corresponds to speed_20 indices 转换为np arrays，由于对内存的关注，我更喜欢使用numpy操作，因此效果很好）

Answer 1

您快到了。这应该做您想要的：

speed_20 = np.where(Speed > 20)[0]
sort = np.argsort(-Brake[speed_20])
result = speed_20[sort[:N]]

Answer 2

使用NumPy，也许这是您可以考虑的选项。

首先创建一个多维矩阵（我更改了值，以使其更容易理解）：

Time =        [  2,   1,   5,   4,   3]
Speed =       [ 10,  20,  40,  30,  50]
Brak_press =  [0.1, 0.3, 0.5, 0.4, 0.2]

data = np.array([Time, Speed, Brak_press]).transpose()

因此数据存储为：

print(data)
# [[ 2.  10.   0.1]
#  [ 1.  20.   0.3]
#  [ 5.  40.   0.5]
#  [ 4.  30.   0.4]
#  [ 3.  50.   0.2]]

要提取大于20的速度：

data[data[:,1] > 20]
# [[ 5.  40.   0.5]
#  [ 4.  30.   0.4]
#  [ 3.  50.   0.2]]

要获得第n大Brak_press：

n = 2
data[data[:,2].argsort()[::-1][:n]]
# [[ 5.  40.   0.5]
#  [ 4.  30.   0.4]]