black-jack 0.1.0

DataFrame / Series / general data processing in Rust
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

//! DataFrame `groupby` functionality.

use num::*;
use std::iter::Sum;

use crate::prelude::*;

/// [`DataFrame::groupby`]  result.
/// Contains the split series by key
pub struct DataFrameGroupBy<T>
where
    T: BlackJackData,
{
    groups: Vec<SeriesGroupBy<T>>,
}

impl<T> DataFrameGroupBy<T>
where
    T: BlackJackData + 'static,
{
    /// Construct a new [`DataFrameGroupBy`] from a collection of [`SeiresGroupBy`]
    /// structs; shouldn't be needed to be used directly.
    pub fn new(groups: Vec<SeriesGroupBy<T>>) -> Self {
        DataFrameGroupBy { groups }
    }

    /// Sum this grouped dataframe object.
    /// basically calls `sum` in parallel on each grouped series collected.
    pub fn sum(&self) -> DataFrame<i32>
    // TODO:
    where
        T: BlackJackData + Copy + Sum + Num + Send + Ord,
    {
        // TODO: Return result

        let mut df = DataFrame::new();

        let _ = self
            .groups
            .iter()
            .map(|series_groupby| series_groupby.sum())
            .map(|series| df.add_column(series).unwrap())
            .collect::<Vec<()>>();
        df
    }
}