c ++犰狳库中的sort_index（）函数给出错误的结果

Question

我在 c++ armadillo 中使用 sort_index() 函数，但它似乎没有给出正确的结果：

输入向量是[3,4,2,1,5] 对两个方向进行排序

arma::sort_index(input, "ascend").print();
arma::sort_index(input, "descend").print();

并获得以下结果： [3,2,0,1,4] 和 [4,1,0,2,3]

两者都不正确。按升序排序应该得到 [2,3,1,0,4] （用numpy.searchsorted 加倍检查并给出上述结果）。

--- 编辑---

感谢您的回复！我现在意识到我误解了sort_index 的索引方式。我还在努力寻找一个高效、等效的函数，在c++中实现np.searchsorted...

Answer 1

犰狳的文档指出

sort_index( X )
sort_index( X, sort_direction )
返回描述 X 元素排序顺序的向量（即它包含 X 元素的索引）

返回的向量对应于原始向量X 中元素的索引，这将产生一个排序的向量。

numpy中对应的函数是argsort而不是searchsorted。事实上，来自 numpy 的 argsort 将提供与犰狳中的 sort_index 相同的解决方案。

在您的示例[3,4,2,1,5] 中，最小元素位于索引 3 中，然后是索引 2 中的元素，然后是 0，依此类推。换句话说，如果你在X 中使用索引[3,2,0,1,4] 的元素，你会得到向量[1, 2, 3, 4, 5]，这是X 排序的。