Python常用统计库Statsmodels介绍

Statsmodels库概述

Statsmodels是一个基于Python的统计模型估计和推断库。它使用NumPy和Pandas库，提供了线性回归、广义线性模型和时间序列模型等多种模型类别。Statsmodels核心理念是利用统计方法对数据进行建模和推断。

基础使用

安装

通过pip可简单安装Statsmodels库。

模型举例

以下是使用最小二乘法进行建模的例子，包括数据生成、模型拟合和结果输出的过程。通过输出结果可以查看模型拟合的优度。

        import numpy as np
        import pandas as pd
        import statsmodels.api as sm
        np.random.seed(1)
        nsample = 100
        x = np.linspace(0, 10, 100)
        X = np.column_stack((x, x ** 2))
        beta = np.array([1, 0.1, 10])
        e = np.random.normal(size=nsample)
        X = sm.add_constant(X)
        y = np.dot(X, beta) + e
        model = sm.OLS(y, X)
        results = model.fit()
        print(results.summary())
        print("Parameters: ", results.params)
        print("R2: ", results.rsquared)
    

应用领域

Statsmodels库在金融、医学研究和经济学等众多领域有广泛应用，比如构建预测模型、风险分析、评估治疗效果等。

更多信息可访问Statsmodels官网：https://www.statsmodels.org/