Matplotlib PCA样品在改变尺寸后不起作用

时间:2015-03-26 21:19:20

标签: python numpy matplotlib pca

我正在尝试学习如何使用matplotlib.mlabPCA。下面我有以下代码:

import numpy as np
from matplotlib import pyplot as plt
from matplotlib.mlab import PCA as mlabPCA
from mpl_toolkits.mplot3d import Axes3D, proj3d

np.random.seed(234234782384239784)

DIMENSIONS = 3

mu_vec1 = np.array([0 for i in xrange(DIMENSIONS)])
cov_mat1 = np.identity(DIMENSIONS)
class1_sample = np.random.multivariate_normal(mu_vec1, cov_mat1, 20).T
assert class1_sample.shape == (DIMENSIONS, 20)

mu_vec2 = np.array([3 for i in xrange(DIMENSIONS)])
cov_mat2 = np.identity(DIMENSIONS)
class2_sample = np.random.multivariate_normal(mu_vec2, cov_mat2, 20).T
assert class2_sample.shape == (DIMENSIONS, 20)

# Combine the two together
all_samples = np.vstack([class1_sample.T, class2_sample.T])
all_samples = all_samples.T
assert all_samples.shape == (DIMENSIONS, 40)

mlab_pca = mlabPCA(all_samples.T)

# 2d plotting
plt.plot(mlab_pca.Y[0:20, 0],
         mlab_pca.Y[0:20, 1],
         'o', markersize=7, color='blue', alpha=0.5, label='class1')
plt.plot(mlab_pca.Y[20:40, 0],
         mlab_pca.Y[20:40, 1],
         '^', markersize=7, color='red', alpha=0.5, label='class2')

plt.xlabel('x_values')
plt.ylabel('y_values')
plt.xlim([-4, 4])
plt.ylim([-4, 4])
plt.legend()
plt.title('Transformed samples with class labels from matplotlib.mlab.PCA()')

plt.show()

如您所见,PCA运行良好,我得到以下图表: enter image description here

但是,当我尝试更改DIMENSIONS = 100时(我正在尝试模拟光谱数据分析),我收到此错误:

RuntimeError: we assume data in a is organized with numrows>numcols

“好的,我可以将PCA应用到此矩阵的转置上。”我天真地告诉自己。

DIMENSIONS = 100
...        
mlab_pca = mlabPCA(all_samples)

plt.plot(mlab_pca.Y[0, 0:20],
         mlab_pca.Y[1, 0:20],
         'o', markersize=7, color='blue', alpha=0.5, label='class1')
plt.plot(mlab_pca.Y[0, 20:40],
         mlab_pca.Y[1, 20:40],
         '^', markersize=7, color='red', alpha=0.5, label='class2')
...    

我的结果看起来完全没了! enter image description here

我做错了吗?或者是说添加许多维度实际上搞乱了我的数据?

1 个答案:

答案 0 :(得分:2)

我不希望这些点分开。 PCA(X)与PCA(X.T)不同.T

似乎需要数量> numcols是matplotlib PCA的限制。 R的prcomp和Python的sklearn PCA都可以使用数字>的矩阵。 numcols或numcols> numRows行。