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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76

extern crate brassfibre;
use brassfibre::prelude::*;

#[test]
fn test_block_sum() {
    let values: Vec<Vec<i64>> = vec![vec![1, 2, 3, 4, 5], vec![6, 7, 8, 9, 10]];
    let index: Vec<i64> = vec![10, 20, 30, 40, 50];
    let columns: Vec<&str> = vec!["X", "Y"];
    let b = Block::from_vec(values, index, columns);

    let exp: Series<i64, &str> = Series::new(vec![15, 40], vec!["X", "Y"]);
    assert_eq!(b.sum(), exp);

    let exp: Series<f64, &str> = Series::new(vec![3., 8.], vec!["X", "Y"]);
    assert_eq!(b.mean(), exp);
}

#[test]
fn test_minmax_int() {
    let values = vec![3, 2, 1, 4, 5, 7, 6, 8, 10, 10, 12, 14, 11, 14, 15];
    let b = Block::from_col_vec(values,
                                vec!["A", "BB", "CC", "D", "EEE"],
                                vec!["X", "YYY", "ZZ"]);
    assert_eq!(b.len(), 5);

    let exp: Series<i64, &str> = Series::new(vec![1, 6, 11], vec!["X", "YYY", "ZZ"]);
    assert_eq!(b.min(), exp);

    let exp: Series<i64, &str> = Series::new(vec![5, 10, 15], vec!["X", "YYY", "ZZ"]);
    assert_eq!(b.max(), exp);
}

#[test]
fn test_minmax_float() {
    let values = vec![3., 2., 1., 4., 5., 7., 6., 8., 10., 10., 12., 14., 11., 14., 15.];
    let b = Block::from_col_vec(values,
                                vec!["A", "BB", "CC", "D", "EEE"],
                                vec!["X", "YYY", "ZZ"]);
    assert_eq!(b.len(), 5);

    let exp: Series<f64, &str> = Series::new(vec![1., 6., 11.], vec!["X", "YYY", "ZZ"]);
    assert_eq!(b.min(), exp);

    let exp: Series<f64, &str> = Series::new(vec![5., 10., 15.], vec!["X", "YYY", "ZZ"]);
    assert_eq!(b.max(), exp);
}

#[test]
fn test_describe_float() {
    let values = vec![3., 2., 1., 4., 5., 7., 6., 8., 10., 10., 12., 14., 11., 14., 15.];
    let b = Block::from_col_vec(values,
                                vec!["A", "BB", "CC", "D", "EEE"],
                                vec!["X", "YYY", "ZZ"]);
    assert_eq!(b.len(), 5);

    let exp_values = vec![5.,
                          3.,
                          1.4142135623730951,
                          1.,
                          5.,
                          5.,
                          8.2,
                          1.6,
                          6.,
                          10.,
                          5.,
                          13.2,
                          1.469693845669907,
                          11.,
                          15.];
    let exp = Block::from_col_vec(exp_values,
                                  vec!["count", "mean", "std", "min", "max"],
                                  vec!["X", "YYY", "ZZ"]);

    assert_eq!(b.describe(), exp);
}