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

Question

我有一个2维的数组y_probability。它包含实例属于每个类的概率。我想创建第一列中从最大到最小排序的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]反转结果{ {1}}返回以升序排序数组的索引：

argsort