本文介绍了如何利用numpy的argmax和argsort函数来确定节点的重要性。通过argmax可以找到最大值对应的节点位置，argsort则能获取前k个最大值的节点位置。在 pagerank 等算法中，这些方法对于提取关键节点至关重要。
在用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个值对应的位置；
扫一扫
5555
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
