从2D numpy数组中获取每个子数组的前N个值

Question

我有一个二维numpy数组（分类模型的概率输出）。现在我想为每个嵌套数组提取前3个值的索引。

E.g。

x = np.array([[1, 2, 3, 5, 6, 8], [4, 5, 6, 9, 10, 1]], np.int32)

我想要的是

[[5, 4, 3]
[4, 3, 2]]

不使用循环