主成分分析python代码实现

版本一:使用numpy、panda

python 复制代码
import numpy as np
import pandas as pd

#PCA:principal component analysis 主成分分析法

#读取数据,以outcome.xlsx为例
data=pd.read_excel('outcome.xlsx',sheet_name='人均消费(元)')
data=data.set_index('省份')

#获取样本数量、指标数量
n,p=data.shape
print(f'样本数量:{n},指标数量:{p}')

#1、Z标准化{消除量纲影响}
data_Z=(data-data.mean(axis=0))/data.std(axis=0)

#2、计算协方差矩阵[相关系数矩阵]
data_cov=data_Z.cov()


#1、2 等价于计算相关系数矩阵
# data_corr=data.corr()

#3、计算相关系数矩阵的特征值与特征向量
eigValues,eigVectors=np.linalg.eig(data_cov)

# 获取特征值的排序索引
sorted_indices = np.argsort(eigValues)[::-1]

# 使用排序索引重新排列特征值和特征向量
sorted_eigValues = eigValues[sorted_indices]
sorted_eigVectors = eigVectors[:, sorted_indices]
#特征向量标准化
sorted_eigVectors = sorted_eigVectors / np.linalg.norm(sorted_eigVectors, axis=0)

#4、计算特征值贡献率与累计贡献率
sorted_eigValues_rate = sorted_eigValues/sorted_eigValues.sum()
sorted_eigVectors_cumrate=sorted_eigValues_rate.cumsum()

print(f'贡献率为   {sorted_eigValues_rate}')
print(f'累计贡献率为  {sorted_eigVectors_cumrate}')

#5、决定主成分数量
m=int(input('选择你需要的主成分数量:'))

#6、获取各主成分中各指标线性组合系数
Reduce_Matrix=sorted_eigVectors[:,0:m]
print(Reduce_Matrix)

版本二:使用sklearn库中的PCA模块

python 复制代码
import numpy as np
import pandas as pd
from sklearn.decomposition import PCA
from sklearn.preprocessing import StandardScaler

# 读取数据,以outcome.xlsx为例
data = pd.read_excel('outcome.xlsx', sheet_name='人均消费(元)')
data = data.set_index('省份')

# 获取样本数量、指标数量
n, p = data.shape
print(f'样本数量: {n}, 指标数量: {p}')

# 1、Z标准化{消除量纲影响}
scaler = StandardScaler()
data_Z = scaler.fit_transform(data)

# 2、使用PCA进行主成分分析
pca = PCA()
pca.fit(data_Z)

# 3、计算特征值贡献率与累计贡献率
explained_variance_ratio = pca.explained_variance_ratio_
cumulative_explained_variance_ratio = np.cumsum(explained_variance_ratio)

print(f'贡献率为   {explained_variance_ratio}')
print(f'累计贡献率为  {cumulative_explained_variance_ratio}')

# 4、决定主成分数量
m = int(input('选择你需要的主成分数量:'))

# 5、获取主成分中各指标线性组合的系数
components = pca.components_.T[:,:m]
# 将系数转换为DataFrame以便更好地查看
components_df = pd.DataFrame(components, index=data.columns, columns=[f'主成分{i+1}' for i in range(m)])
print(components_df)

# 6、计算降维后的矩阵并输出
pca = PCA(n_components=m)
Reduce_Matrix = pca.fit_transform(data_Z)
col = [f'主成分{i+1}' for i in range(m)]
New_outcome = pd.DataFrame(data=Reduce_Matrix, index=data.index, columns=col)
print(New_outcome)
相关推荐
2401_8582861110 分钟前
CD21.【C++ Dev】类和对象(12) 流插入运算符的重载
开发语言·c++·算法·类和对象·运算符重载
橘猫云计算机设计11 分钟前
基于ssm的食物营养成分数据分析平台设计与实现(源码+lw+部署文档+讲解),源码可白嫖!
后端·python·信息可视化·数据挖掘·数据分析·django·毕业设计
上理考研周导师21 分钟前
【虚拟仪器技术】Labview虚拟仪器技术应用教程习题参考答案[13页]
服务器·开发语言
郭涤生33 分钟前
Chapter 5: The Standard Library (C++20)_《C++20Get the details》_notes
开发语言·c++·笔记·c++20
叶孤程1 小时前
Qt图形化界面为何总被“冷落“?
开发语言·c++·qt
葵野寺1 小时前
【多线程】线程池
java·开发语言·java-ee·web app
高林雨露1 小时前
Java 与 Kotlin 对比学习指南(二)
java·开发语言·kotlin
笑口常开xpr1 小时前
C 语 言 --- 整 形 提 升
c语言·开发语言
liuhaoran___1 小时前
计算机求职面试中高频出现的经典题目分类整理
python
martian6652 小时前
Maven核心配置文件深度解析:pom.xml完全指南
java·开发语言