python - python 中数据的多维缩放

Question

我有以下代码将多维缩放应用于名为 parkinsonD​​ata 的数据样本:

iterations=4
count=0
while(count<iterations):
    mds1=manifold.MDS(n_components=2, max_iter=3000)
    pos=mds1.fit(parkinsonData).embedding_
    plt.scatter(pos[:, 0], pos[:, 1])
    count=count+1

这样，我得到了该 MDS 算法的 4 个不同的图，由于随机种子的原因，所有图都不同。这些图具有不同的颜色，但 parkinsonD​​ata 有一个名为 status 的列，其中包含 0 或 1 个值，我想在每个图中用不同的颜色显示这种差异。

例如我想实现:

一个绘图，状态字段中的 0 值使用一种颜色，状态字段中的 1 值使用不同的颜色。

第二个图，状态字段中的 0 值使用一种颜色，状态字段中的 1 值使用不同的颜色。 (两种颜色都与第一个图不同)

第三个图，状态字段中的 0 值使用一种颜色，状态字段中的 1 值使用不同的颜色。 (两种颜色都与第一图和第二图不同)

第四个图，状态字段中的 0 值使用一种颜色，状态字段中的 1 值使用不同的颜色。 (两种颜色与第一、第二和第三图不同)

有人知道如何实现这种预期行为吗？

Answer 1

你可以做这样的事情

%matplotlib inline
import matplotlib.pyplot as plt

# example data
Y = [[ 1 , 2 , 3 ,6], [ 1 , 2 , 3 ,6], [ 1 , 2 , 3 ,6], [ 1 , 2 , 3 ,6]]
X = [[ 1 , 2 , 4 ,5], [ 1 , 2 , 3 ,6], [ 1 , 2 , 3 ,6], [ 1 , 2 , 3 ,6]]
status = [[0,1,0,0], [0,0,1,1], [1,1,0,0], [0,1,0,1]]

# create a list of list of unique colors for 4 plots
my_colors = [['red','green'],['blue','black'],['magenta','grey'],['purple','cyan']]


iterations=4
count=0
while(count<iterations):
    plt.figure()
    for i,j in enumerate(X):
        plt.scatter(X[count][i],Y[count][i],color = my_colors[count][status[count][i]])
    count=count+1
    plt.show()

结果(我只附加 2 个图像，但 4 个图像是使用独特的颜色集创建的)