numpy中是否有任何函数可以获取数组的前k个索引和值？

Question

我有一个数组，我想得到两个大小为k的数组，其中包含该数组的前k个值和相应的索引？

我们在Tensorflow中具有这样的功能。 tf.nn.top_k

假设数组是a=np.array([4,5,6,7,2,3,9])

values, indexes = function(a, 3)应该返回

values = array([9,7,6])
indexes = array([6,3,2])