#traits #back-end #function #vec #datatype #analysis #ndarray

tea-dtype

提供数据类型和有用特质的库

10个版本

0.3.3 2024年8月21日
0.3.2 2024年8月19日
0.2.6 2024年8月12日
0.2.5 2024年7月24日

318财务

Download history 457/week @ 2024-07-12 164/week @ 2024-07-19 40/week @ 2024-07-26 2/week @ 2024-08-02 304/week @ 2024-08-09 405/week @ 2024-08-16

780 每月下载量
用于 5 个crate(2个直接使用)

MIT 许可证

46KB
1.5K SLoC

Tevec

Build Crates.io Version

简介

一个crate,用于在不同的后端(目前为 Vec & Ndarray & Polars)之间提供财务量化分析函数。几乎完全使用Rust traits 实现,以方便未来支持更多后端。

目前,它主要分为三个功能类别

  • 滚动函数
  • 映射函数
  • 聚合函数

安装

tevec = "0.2" 添加到您的 Cargo.toml

基本用法

首先,导入常见的trait名称以调用相应的方法。 use tevec::prelude::*

聚合函数

大多数聚合函数都是为满足 IntoIterator + Sized trait 的struct实现的。

let data = vec![1, 2, 3, 4, 5];
data.titer().mean();  // not consume data, return Some(3)
data.mean();  // consume data, return Some(3)
let data = vec![1., f64::NAN, 3.];
data.titer().vmean();  // valid mean, this will ignore nan, return 2.
// valid function can also be used for Option<T> dtype
let data = vec![Some(1), None, Some(3)]
data.vmean(); // return 2.

使用 titer 返回一个满足 IteratorTrustedLen,允许进行进一步的方法调用。 titer 方法来自 Titer trait,该trait已为所有后端实现。

滚动函数

let data = vec![1, 2, 3, 4, 5];
let mean: Vec<f64> = data.ts_mean(3, Some(1)); // params: window, min_periods
let mean2: Array1<f32> = data.ts_vmean(4, None); // rolling_mean function ignore none values

映射函数

 let v = vec![1., 2., 3., 4., 5.];
 let shift_v: Vec<_> = v.titer().vshift(2, None).collect_trusted_vec1();
 let shfit_abs_v: Vec<_> = v.titer().abs().vshift(2, None).collect_trusted_vec1();

一些映射函数返回一个 Iterator,允许在不重新分配内存的情况下进行链式调用,并且只在需要时收集迭代器。

特性标志

pl:对于 Polars 后端

ndarray:对于 Ndarray 后端

agg:聚合函数

map:映射函数

rolling:滚动函数

stat: 统计函数

time: DateTimeTimeDelta 结构体

依赖项

~94–550KB
~10K SLoC