tsflex: 灵活高效的时间序列处理与特征提取工具包

tsflex简介

tsflex是一个强大的Python工具包,专门用于时间序列的处理和特征提取。它的设计理念是灵活性和高效性,能够适应各种复杂的时间序列数据分析需求。无论是多变量时间序列、不规则采样数据,还是具有多种窗口和步长配置的特征提取,tsflex都能轻松应对。

tsflex logo

主要特性

灵活性

tflex的一大亮点就是其灵活性:

支持处理多变量/多模态时间序列数据
集成了多个流行的数据处理和特征提取库,如scipy、numpy、statsmodels等
特征提取支持多种窗口大小和步长配置
对数据采样率和序列对齐没有严格要求,可处理异步数据

这种灵活性使得tsflex能够适应各种复杂的时间序列分析场景。

高效性

tflex在设计时特别注重性能:

采用基于视图的操作进行数据处理和特征提取
极低的内存峰值使用
快速的执行速度

官方提供的基准测试显示,tsflex在性能上优于同类工具包。

直观易用

tflex提供了直观的API:

保留数据的序列索引
特征提取结果具有可解释的列名
API设计简洁明了

这些特性大大降低了使用门槛,使得研究人员和数据科学家能够快速上手。

安装与使用

安装tsflex非常简单,可以通过pip或conda进行:

pip install tsflex

或

conda install -c conda-forge tsflex

下面是一个简单的特征提取示例:

import pandas as pd
import numpy as np
import scipy.stats as ss
from tsflex.features import MultipleFeatureDescriptors, FeatureCollection
from tsflex.utils.data import load_empatica_data

# 1. 加载序列索引数据(这里是时间索引)
df_tmp, df_acc, df_ibi = load_empatica_data(['tmp', 'acc', 'ibi'])

# 2. 构建特征提取配置
fc = FeatureCollection(
    MultipleFeatureDescriptors(
        functions=[np.min, np.mean, np.std, ss.skew, ss.kurtosis],
        series_names=["TMP", "ACC_x", "ACC_y", "IBI"],
        windows=["15min", "30min"],
        strides="15min",
    )
)

# 3. 提取特征
fc.calculate(data=[df_tmp, df_acc, df_ibi], approve_sparsity=True)

这个例子展示了tsflex如何处理具有不同采样率的多变量时间序列数据。