numpy中argmax和argsort用法

在用pagerank等算法计算出每个节点的重要性后，我们一般需要提取前k个最大节点信息。

测试数据为

test_data = np.array([1, 0.5, 3.2, 1.7, 2.1, 2, 3, 1.5, 1.6, 2.7])

用numpy.argmax()求出最大节点信息

> np.argmax(test_data)
2  # 对应的值是3.2

用numpy.argsort()求出前k个最大节点信息

>k = 3
>np.argsort(-test_data)[:k]
array([2, 6, 9])  # 对应的值是3.2，3， 2.7

总结

用np.argmax方法提取最大值对应的位置；

用np.argsort方法提取前k个值对应的位置；