stock-trek 0.2.3

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

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

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

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

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

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