Nick
2020-11-28 12:25:13
If you want the three largest of the four values, then you can just find all the values that are greater than the minimum value. You can use argwhere to get just an array of indexes:
import numpy as np
pa = np.random.normal(10, 5, 4)
ml = np.min(pa)
mll = np.argwhere(pa > ml)
print(mll)
Sample output:
[[0]
[1]
[3]]
To flatten that output, convert to an array and flatten:
mll = np.array(np.nonzero(pa > ml)).flatten()
print(mll)
Sample output:
[0 1 3]
热门github
6
@Zhengrong see my edit