polars-ta 0.1.0

Polars Series interface for technical indicators — native Polars integration, aligned with ta-lib
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

use polars_core::prelude::*;
use polars_ta_core::trend::{
    dema as dema_core, ema as ema_core, tema as tema_core, wma as wma_core,
};

fn series_to_f64(s: &Series) -> PolarsResult<Vec<f64>> {
    Ok(s.cast(&DataType::Float64)?
        .f64()?
        .into_iter()
        .map(|v| v.unwrap_or(f64::NAN))
        .collect())
}

pub fn ema_series(close: &Series, period: usize) -> PolarsResult<Series> {
    let data = series_to_f64(close)?;
    let result = ema_core(&data, period);
    Ok(Float64Chunked::from_vec("ema".into(), result).into_series())
}

pub fn wma_series(close: &Series, period: usize) -> PolarsResult<Series> {
    let data = series_to_f64(close)?;
    let result = wma_core(&data, period);
    Ok(Float64Chunked::from_vec("wma".into(), result).into_series())
}

pub fn dema_series(close: &Series, period: usize) -> PolarsResult<Series> {
    let data = series_to_f64(close)?;
    let result = dema_core(&data, period);
    Ok(Float64Chunked::from_vec("dema".into(), result).into_series())
}

pub fn tema_series(close: &Series, period: usize) -> PolarsResult<Series> {
    let data = series_to_f64(close)?;
    let result = tema_core(&data, period);
    Ok(Float64Chunked::from_vec("tema".into(), result).into_series())
}