1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20

use crate::prelude::*;
use arrow::chunk::Chunk;

pub type ArrowChunk = Chunk<ArrayRef>;

impl std::convert::TryFrom<(ArrowChunk, &[ArrowField])> for DataFrame {
    type Error = PolarsError;

    fn try_from(arg: (ArrowChunk, &[ArrowField])) -> Result<DataFrame> {
        let columns: Result<Vec<Series>> = arg
            .0
            .columns()
            .iter()
            .zip(arg.1)
            .map(|(arr, field)| Series::try_from((field.name.as_ref(), arr.clone())))
            .collect();

        DataFrame::new(columns?)
    }
}