PyCorrAna - Python Correlation Analysis Toolkit

自动化相关性分析工具 - 降低决策成本，一键输出关键结果

特性 • 安装 • 快速开始 • 文档 • 示例

初心

相关性的分析，在实际分析工作中，并不是最复杂的部分，却常常在入门阶段挡住很多人的学习/工作进度， 这是我自己常用的一些工具的重新包装，希望能让更多人可以快速上手相关性分析。

特性

PyCorrAna 是一个方便快速入手的 Python 相关性分析工具，核心设计理念：

• 自动化常规操作 - 智能识别数据类型，自动选择最优相关系数方法
• 降低决策成本 - 无需纠结用 Pearson 还是 Spearman，工具自动帮你选择
• 一键输出关键结果 - 从数据加载到结果导出，一行代码搞定

主要功能

安装

# 基础安装
pip install pycorrana

# 包含所有可选依赖
pip install pycorrana[all]

# 开发模式安装
git clone https://github.com/sidneylyzhang/pycorrana.git
cd pycorrana
pip install -e .

另外，我个人建议还是使用 uv 进行安装，可以获得更好的使用体验，也不会过多影响当前的python环境，也可以更方便的使用工具自带的两个命令行工具。

快速开始

1. 一行代码完成分析

from pycorrana import quick_corr

# 从文件加载并分析
result = quick_corr('data.csv')

# 或者使用 DataFrame
import pandas as pd
df = pd.read_csv('data.csv')
result = quick_corr(df, target='sales')

2. 交互式分析

# 启动交互式命令行工具
pycorrana-interactive

3. 命令行工具

# 完整分析
pycorrana analyze data.csv --target sales --export results.xlsx

# 数据清洗
pycorrana clean data.csv --dropna --output cleaned.csv

# 偏相关分析
pycorrana partial data.csv -x income -y happiness -c age,education

# 非线性依赖检测
pycorrana nonlinear data.csv --top 20

详细示例

示例 1: 基础相关性分析

from pycorrana import quick_corr
from pycorrana.datasets import load_iris

# 加载示例数据
df = load_iris()

# 一键分析
result = quick_corr(
    df,
    method='auto',           # 自动选择方法
    missing_strategy='warn', # 缺失值警告但不处理
    plot=True,               # 自动生成图表
    export='results.xlsx',   # 导出结果
    verbose=True             # 输出详细信息
)

# 查看结果
corr_matrix = result['correlation_matrix']
pvalue_matrix = result['pvalue_matrix']
significant_pairs = result['significant_pairs']

示例 2: 指定目标变量

from pycorrana import quick_corr

# 只分析与目标变量的相关性
result = quick_corr(
    df,
    target='survived',  # 目标变量
    method='auto',
    plot=True
)

# 查看与目标变量最相关的特征
for pair in result['significant_pairs'][:5]:
    print(f"{pair['var1']} vs {pair['var2']}: {pair['correlation']:.3f}")

示例 3: 使用分析器类（更灵活）

from pycorrana import CorrAnalyzer

# 创建分析器
analyzer = CorrAnalyzer(df, method='auto', verbose=True)

# 执行分析
analyzer.fit()

# 自定义可视化
analyzer.plot_heatmap(cluster=True, savefig='heatmap.png')
analyzer.plot_pairplot(columns=['A', 'B', 'C', 'D'])

# 导出结果
analyzer.export_results('results.xlsx', format='excel')

# 查看文本摘要
print(analyzer.summary())

示例 4: 大数据集优化

from pycorrana import CorrAnalyzer
from pycorrana.utils import LargeDataConfig

# 配置大数据优化参数
config = LargeDataConfig(
    sample_size=100000,      # 采样大小
    auto_sample=True,        # 自动采样
    auto_optimize=True,      # 自动优化内存
    verbose=True
)

# 使用配置分析大数据集
analyzer = CorrAnalyzer(large_df, large_data_config=config)
analyzer.fit()

示例 5: 偏相关分析

from pycorrana import partial_corr, partial_corr_matrix, semipartial_corr

# 控制年龄后，计算收入与幸福感的偏相关
result = partial_corr(
    df,
    x='income',
    y='happiness',
    covars=['age', 'education'],  # 控制变量
    method='pearson'
)

print(f"偏相关系数: {result['partial_correlation']:.3f}")
print(f"p值: {result['p_value']:.4e}")
print(f"95% CI: [{result['ci_95'][0]:.3f}, {result['ci_95'][1]:.3f}]")

# 半偏相关（部分相关）
result = semipartial_corr(df, x='income', y='happiness', covars='age')

# 偏相关矩阵
pcorr_matrix = partial_corr_matrix(df, covars='age')
print(pcorr_matrix)

示例 6: 非线性依赖检测

from pycorrana import distance_correlation, mutual_info_score, maximal_information_coefficient
from pycorrana.core.nonlinear import nonlinear_dependency_report

# 距离相关（可检测非线性关系）
result = distance_correlation(df['X'], df['Y'], return_pvalue=True)
print(f"dCor: {result['dcor']:.3f}, p-value: {result['p_value']:.4f}")

# 互信息
result = mutual_info_score(df['X'], df['Y'])
print(f"MI: {result['mi_normalized']:.3f}")

# 最大信息系数 (MIC)
# 注意：当前 MIC 为纯 Python 实现，计算速度较慢
# 对于大数据集，建议先采样：smart_sample(df, sample_size=500)
result = maximal_information_coefficient(df['X'], df['Y'])
print(f"MIC: {result['mic']:.3f}")

# 生成非线性检测报告
report = nonlinear_dependency_report(df, top_n=10)
print(report)

MIC 性能说明：当前版本的 MIC 实现为纯 Python 编写，没有特殊优化，计算速度较慢。对于大数据集（n > 1000），建议先使用 smart_sample 进行采样处理。

示例 7: 数据清洗和预处理

from pycorrana.utils.data_utils import load_data, handle_missing, detect_outliers

# 加载数据
df = load_data('data.csv')  # 自动识别格式

# 缺失值处理
df_clean = handle_missing(
    df,
    strategy='fill',      # 填充策略
    fill_method='knn',    # KNN预测填充
    verbose=True
)

# 异常值检测
outliers = detect_outliers(
    df,
    method='iqr',         # IQR方法
    visualize=True        # 显示箱线图
)

示例 8: 典型相关分析 (CCA)

from pycorrana import cca, cca_permutation_test, CCAAnalyzer

# 典型相关分析：研究两组变量之间的关系
# 例如：身体指标（身高、体重）与运动能力（速度、力量）
X_vars = ['height', 'weight', 'age']
Y_vars = ['speed', 'strength', 'endurance']

result = cca(
    df[X_vars], 
    df[Y_vars],
    compute_significance=True,
    confidence_level=0.95
)

# 查看典型相关系数
print("典型相关系数:", result['canonical_correlations'])

# 查看显著性检验结果
for test in result['significance_tests']:
    print(f"第{test['canonical_index']}典型相关: r={test['canonical_correlation']:.3f}, p={test['p_value']:.4f}")

# 查看冗余指数（解释方差比例）
print("X被Y解释的冗余:", result['redundancy_X_given_Y'])
print("Y被X解释的冗余:", result['redundancy_Y_given_X'])

# 置换检验（更稳健的显著性验证）
perm_result = cca_permutation_test(df[X_vars], df[Y_vars], n_permutations=1000)
print(f"置换检验 p 值: {perm_result['p_value']:.4f}")

自动方法选择规则

PyCorrAna 根据变量类型自动选择最优的相关系数方法：

项目结构

pycorrana/
├── src/pycorrana/            # 源代码
│   ├── __init__.py           # 包入口
│   ├── core/                 # 核心分析模块
│   │   ├── analyzer.py       # 主分析器
│   │   ├── visualizer.py     # 可视化
│   │   ├── reporter.py       # 报告生成
│   │   ├── partial_corr.py   # 偏相关分析
│   │   ├── nonlinear.py      # 非线性检测
│   │   └── cca.py            # 典型相关分析
│   ├── utils/                # 工具函数
│   │   ├── data_utils.py     # 数据处理
│   │   ├── stats_utils.py    # 统计工具
│   │   └── large_data.py     # 大数据优化
│   ├── cli/                  # 命令行工具
│   │   ├── main_cli.py       # 主CLI
│   │   └── interactive.py    # 交互式CLI
│   └── datasets.py           # 示例数据集
├── tests/                    # 测试
├── examples/                 # 示例代码
├── docs/                     # 文档
├── pyproject.toml            # 项目配置
└── README.md                 # 本文件

依赖

• Python >= 3.10
• numpy >= 1.21.0
• pandas >= 1.3.0
• scipy >= 1.7.0
• matplotlib >= 3.5.0
• seaborn >= 0.11.0
• statsmodels >= 0.13.0
• scikit-learn >= 1.0.0
• polars >= 0.15.0
• typer >= 0.9.0
• rich >= 13.0.0
• openpyxl >= 3.0.0

许可证

MIT License - 详见 LICENSE 文件

贡献

欢迎提交 Issue 和 Pull Request！