如何根据其中一个Vecs中的值对两个Vecs进行共同排序?

时间:2015-09-14 12:10:25

标签: rust

我有两个Vec对应于一个特征向量列表及其对应的类标签,我想按类别标签对它们进行共同排序。

但是,Rust sort_by在一个切片上操作而不是在特征(或类似)上的泛型函数,并且闭包只获得要比较的元素而不是索引,所以我可以偷偷摸摸地破坏那种平行。

我考虑过这个解决方案:

let mut both = data.iter().zip(labels.iter()).collect();
both.sort_by( blah blah );
// Now split them back into two vectors

我不希望每次都分配一个全新的矢量,因为数据的大小可能非常大。

当然,我总是可以实现自己的排序,但如果有一种内置的方法可以做到这一点,那就更好了。

1 个答案:

答案 0 :(得分:4)

我刚写了a crate "permutation",允许你这样做:)

let names = vec!["Bob", "Steve", "Jane"];
let salary = vec![10, 5, 15];
let permutation = permutation::sort(&salary[..]);
let ordered_names = permutation.apply_slice(&names[..]);
let ordered_salaries = permutation.apply_slice(&salary[..]);
assert!(ordered_names == vec!["Steve", "Bob", "Jane"]);
assert!(ordered_salaries == vec![5, 10, 15]);

将来可能会在单个函数调用中支持这一点。