polars-mem-engine 0.54.2

In memory engine of the Polars project.
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

use polars_ops::frame::gather::GatherDf;
use recursive::recursive;

use super::*;

pub struct GatherExec {
    input: Box<dyn Executor>,
    idxs: Box<dyn Executor>,
    null_on_oob: bool,
}

impl GatherExec {
    pub fn new(input: Box<dyn Executor>, idxs: Box<dyn Executor>, null_on_oob: bool) -> Self {
        GatherExec {
            input,
            idxs,
            null_on_oob,
        }
    }
}

impl Executor for GatherExec {
    #[recursive]
    fn execute<'a>(&'a mut self, state: &'a mut ExecutionState) -> PolarsResult<DataFrame> {
        let input = self.input.execute(state)?;
        let idxs = self.idxs.execute(state)?;
        assert!(idxs.width() == 1);
        input.gather_with_column(&idxs.columns()[0], self.null_on_oob)
    }
}