创建从最大到最小的数组索引列表

Question

我有一个数组y_probability，该数组为2维。它包含一个实例属于每个类的概率。我想创建一个从第一列中从最大到最小的y_probabiltiy数组的索引列表。我怎样才能做到这一点？

样本数据：

y_probability = np.array([[0.3,0.7],[0.5,0.5] ,[0.2,0.8], [0.1,0.9]])

所需的输出：

index = [2,1,3,4]

Answer 1

您可以在第一列上使用 argsort(请注意，python中的索引为0，因此结果将比您预期的少于一个(，然后用[::-1]自argsort扭转结果。返回按上升顺序排列数组的索引：

y_probability[:,0].argsort()[::-1]
# array([1, 0, 2, 3])