Python中如何处理高维数据—PCA降维实战案例-Python教程-PHP中文网

Python中如何处理高维数据—PCA降维实战案例

雪夜

发布： 2025-07-13 14:00:03

原创

139人浏览过

pca（主成分分析）是一种通过线性投影降低数据维度的方法，能保留最大方差信息以减少冗余和计算复杂度。1. 其核心思想是提取正交的主成分来捕捉数据主要变化方向；2. 适用于高维场景如图像、文本处理；3. 实战步骤包括：导入数据、标准化、应用pca降维、可视化结果；4. 选择主成分数量可通过解释方差比或累计曲线判断；5. 注意事项有：需标准化、不适用于非线性结构与分类特征选择、可能损失有用信号。

Python中如何处理高维数据—PCA降维实战案例

在Python中处理高维数据时，PCA（主成分分析）是一种非常实用的降维方法。它能帮助我们减少特征数量，同时保留尽可能多的信息。下面通过一个实战案例，带你了解如何用PCA进行降维。

什么是PCA？为什么适合用来处理高维数据？

PCA 的核心思想是将原始特征空间中的信息，投影到一个更低维度的空间中，从而提取出最重要的几个“主成分”。这些主成分之间相互正交，能最大程度地保留原始数据的方差信息。

高维数据的问题在于计算复杂度高、容易过拟合，而且很多特征之间可能存在冗余。PCA 就能有效解决这些问题，尤其适用于图像、文本等特征维度动辄成百上千的场景。

立即学习“Python免费学习笔记（深入）”；

实战步骤：使用 sklearn 实现 PCA

我们以经典的鸢尾花（Iris）数据集为例，虽然它的维度并不算高（4个特征），但作为入门练习非常合适。

第一步：导入必要的库和数据

from sklearn.datasets import load_iris
from sklearn.decomposition import PCA
import matplotlib.pyplot as plt

# 加载数据
iris = load_iris()
X = iris.data
y = iris.target

登录后复制

第二步：标准化数据

PCA 对数据尺度敏感，所以需要先做标准化：

from sklearn.preprocessing import StandardScaler

scaler = StandardScaler()
X_scaled = scaler.fit_transform(X)

登录后复制

第三步：应用 PCA 进行降维

这里我们尝试降到2维，方便可视化：

pca = PCA(n_components=2)
X_pca = pca.fit_transform(X_scaled)

登录后复制

第四步：可视化结果

plt.figure(figsize=(8, 6))
for target in [0, 1, 2]:
    plt.scatter(X_pca[y == target, 0], X_pca[y == target, 1], label=iris.target_names[target])
plt.xlabel('PC1')
plt.ylabel('PC2')
plt.legend()
plt.title('PCA of Iris Dataset')
plt.show()

登录后复制

这样我们就完成了整个流程，可以看到不同类别的点被较好地区分开。

如何选择主成分数量？

这是使用 PCA 时最常遇到的问题之一。你可以通过查看解释方差比来决定保留多少主成分：

pca = PCA()
pca.fit(X_scaled)
explained_variance = pca.explained_variance_ratio_
print(explained_variance)

登录后复制

输出类似：

[0.729, 0.228, 0.036, 0.007]

登录后复制

这说明前两个主成分已经解释了大约 95.7% 的信息，因此可以放心地只保留前两个成分。

也可以画出累计解释方差曲线来找拐点：

import numpy as np

cumulative_variance = np.cumsum(explained_variance)
plt.plot(cumulative_variance)
plt.xlabel('Number of components')
plt.ylabel('Cumulative explained variance')
plt.grid()
plt.show()

登录后复制