sklearn PCA的使用

import numpy as np
from sklearn.decomposition import PCA

# 训练数据
train_data = np.array([[1, 2, 3], [4, 8, 12], [16, 32, 48]])
# 构造PCA实例，n_components：目标维度；whiten：是否白化
pca = PCA(n_components=2, whiten=True)
# 使用数据训练PCA
pca.fit(train_data)
# 数据降维
source = np.array([[1, 2, 3], [6, 6, 6], [1, 2, 3]])
transformed = pca.transform(source)
print(transformed)