brassfibre 0.2.0

Provides multiple-dtype columner storage, known as DataFrame in pandas/R
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59

use std::hash::Hash;

use super::Indexer;
use nullvec::prelude::dev::algos::Sorter;

//**********************************************
//*Soat
//**********************************************

impl<U> Indexer<U>
    where U: Clone + Eq + Hash + Ord
{
    pub fn argsort(&self) -> (Vec<usize>, Self) {
        let (indexer, sorted) = Sorter::argsort(&self.values);
        let sorted = Indexer::new(sorted);
        (indexer, sorted)
    }

    pub fn sort(&self) -> Self {
        Indexer::new(Sorter::sort(&self.values))
    }
}


#[cfg(test)]
mod tests {

    use super::super::Indexer;

    #[test]
    fn test_index_argsort_int() {
        let idx = Indexer::new(vec![5, 4, 3, 2, 1]);
        let (indexer, sorted) = idx.argsort();
        assert_eq!(indexer, vec![4, 3, 2, 1, 0]);
        assert_eq!(sorted, Indexer::new(vec![1, 2, 3, 4, 5]));
    }

    #[test]
    fn test_index_sort_int() {
        let idx = Indexer::new(vec![5, 4, 3, 2, 1]);
        let sorted = idx.sort();
        assert_eq!(sorted, Indexer::new(vec![1, 2, 3, 4, 5]));
    }

    #[test]
    fn test_index_argsort_str() {
        let idx = Indexer::new(vec!["d", "b", "a", "c"]);
        let (indexer, sorted) = idx.argsort();
        assert_eq!(indexer, vec![2, 1, 3, 0]);
        assert_eq!(sorted, Indexer::new(vec!["a", "b", "c", "d"]));
    }

    #[test]
    fn test_index_sort_str() {
        let idx = Indexer::new(vec!["d", "b", "a", "c"]);
        let sorted = idx.sort();
        assert_eq!(sorted, Indexer::new(vec!["a", "b", "c", "d"]));
    }
}