Question

我有一个约1500行的数据集，有6个变量，我用PCA处理并显示：

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

data = np.array(data)

    try:
        results = PCA(data)
    except:
        raise

    #this will return an array of variance percentages for each component
    print results.fracs

    #
    print results.Wt

    #this will return a 2d array of the data projected into PCA space
    print results.Y 

    x = []
    y = []
    z = []
    for item in results.Y:
        x.append(item[0])
        y.append(item[1])
        z.append(item[2])

    fig1 = plt.figure() # Make a plotting figure
    ax = Axes3D(fig1) # use the plotting figure to create a Axis3D object.
    pltData = [x,y,z] 
    ax.scatter(pltData[0], pltData[1], pltData[2], 'bo') # make a scatter plot of blue dots from the data

    # make simple, bare axis lines through space:
    xAxisLine = ((min(pltData[0]), max(pltData[0])), (0, 0), (0,0)) # 2 points make the x-axis line at the data extrema along x-axis 
    ax.plot(xAxisLine[0], xAxisLine[1], xAxisLine[2], 'r') # make a red line for the x-axis.
    yAxisLine = ((0, 0), (min(pltData[1]), max(pltData[1])), (0,0)) # 2 points make the y-axis line at the data extrema along y-axis
    ax.plot(yAxisLine[0], yAxisLine[1], yAxisLine[2], 'r') # make a red line for the y-axis.
    zAxisLine = ((0, 0), (0,0), (min(pltData[2]), max(pltData[2]))) # 2 points make the z-axis line at the data extrema along z-axis
    ax.plot(zAxisLine[0], zAxisLine[1], zAxisLine[2], 'r') # make a red line for the z-axis.

    # label the axes 
    ax.set_xlabel("x-axis label") 
    ax.set_ylabel("y-axis label")
    ax.set_zlabel("y-axis label")
    ax.set_title("The title of the plot")

    plt.show() # show the plot

我现在想要的是，例如，如何根据另一个数据变量为颜色显示的点着色。例如，如果我在['blue'，'red'，'green']的范围内添加一个名为color的变量到每一行数据，我可以用它来显示这些点的颜色吗？

Answer 1

我认为最简单的方法是，如你所说，每个数据点都有颜色列表= ['blue'，'red'，'green'，...]，然后绘制每个点单独：

for i,xi in enumerate(x):
    ax.scatter(x[i],y[i],z[i],color=colors[i])

另一种方法是查看this question。在这里，您可以为每个在颜色图上显示为颜色的点指定一个数字。所以你可以显示给定参数的渐变。

编辑：要回答您的评论，可以将颜色指定为0到1之间的3个值（RGB）的元组

from matplotlib.colors import Normalize
xnorm=Normalize(x.min(),x.max())
ynorm=Normalize(y.min(),y.max())
colors=[(xnorm(x[i]),ynorm(y[i]),0) for i in range(x.size)]
ax.scatter(x,y,c=colors)

使用Matplotlib进行主成分分析

1 个答案: