0

PCAで処理して表示する6つの変数を持つ〜1500行のデータセットがあります。

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 という変数を追加すると、それを使用して点の色を表示できますか?

4

1 に答える 1

1

最も簡単な方法は、すべてのデータポイントの色のリストcolors = ['blue'、'red'、'green'、...]を用意してから、各ポイントを個別にプロットすることです。

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

別の方法は、この質問を確認することです。ここでは、カラーマップ上で色として表されるすべてのポイントに番号を割り当てることができます。したがって、特定のパラメーターの勾配を表示できます。

編集：あなたのコメントに答えるために、色は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)
于 2013-03-05T10:44:00.947 に答える