stock-trek 0.2.2

Stock Trek time-series analysis
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

#[cfg(feature = "python")]
use pyo3::{exceptions::PyRuntimeError, pyfunction, PyResult};

#[cfg(feature = "python")]
#[pyfunction]
pub fn lag(values: Vec<f64>, lag: usize) -> PyResult<Vec<Option<f64>>> {
    let result = crate::statistics::transformations::lag(&values, lag)
        .map_err(|e| PyRuntimeError::new_err(e.to_string()))?;
    Ok(result)
}

#[cfg(feature = "python")]
#[pyfunction]
pub fn difference(values: Vec<f64>, order: usize) -> PyResult<Vec<f64>> {
    let result = crate::statistics::transformations::difference(&values, order)
        .map_err(|e| PyRuntimeError::new_err(e.to_string()))?;
    Ok(result)
}

#[cfg(feature = "python")]
#[pyfunction]
pub fn seasonal_difference(values: Vec<f64>, period: usize) -> PyResult<Vec<f64>> {
    let result = crate::statistics::transformations::seasonal_difference(&values, period)
        .map_err(|e| PyRuntimeError::new_err(e.to_string()))?;
    Ok(result)
}

#[cfg(feature = "python")]
#[pyfunction]
pub fn logarithm(values: Vec<f64>) -> PyResult<Vec<f64>> {
    let result = crate::statistics::transformations::logarithm(&values)
        .map_err(|e| PyRuntimeError::new_err(e.to_string()))?;
    Ok(result)
}

#[cfg(feature = "python")]
#[pyfunction]
pub fn box_cox(values: Vec<f64>, lambda: f64) -> PyResult<Vec<f64>> {
    let result = crate::statistics::transformations::box_cox(&values, lambda)
        .map_err(|e| PyRuntimeError::new_err(e.to_string()))?;
    Ok(result)
}

#[cfg(feature = "python")]
#[pyfunction]
pub fn detrend_linear(values: Vec<f64>) -> PyResult<Vec<f64>> {
    let result = crate::statistics::transformations::detrend_linear(&values)
        .map_err(|e| PyRuntimeError::new_err(e.to_string()))?;
    Ok(result)
}

#[cfg(feature = "python")]
#[pyfunction]
pub fn rolling_mean(values: Vec<f64>, window: usize) -> PyResult<Vec<f64>> {
    let result = crate::statistics::transformations::rolling_mean(&values, window)
        .map_err(|e| PyRuntimeError::new_err(e.to_string()))?;
    Ok(result)
}

#[cfg(feature = "python")]
#[pyfunction]
pub fn rolling_standard_deviation(values: Vec<f64>, window: usize) -> PyResult<Vec<f64>> {
    let result = crate::statistics::transformations::rolling_standard_deviation(&values, window)
        .map_err(|e| PyRuntimeError::new_err(e.to_string()))?;
    Ok(result)
}