巨蟒散点图

Question

我有一个大小为100x2的向量X和长度为100的向量y ={1，-1}中相应的二进制标签。我想用s.t绘制散乱的数据。我在轴上得到特征，数据点的颜色对应于标签，例如红色是-1，黄色是1表示给定的数据点。

我一直在研究matplotlib和fcn分散，但是它只接受一个特征向量及其标签。

如果能提供任何帮助，我将不胜感激。

Answer 1

您可以使用seaborn (或者matplotlib )轻松地做到这一点。下面是密码。

1. ，我正在创建一个大小为100x2的随机数组，并将其命名为X。我正在创建一个大小为100 x1的0和1的随机数组，并将其称为Y

。

>> import numpy as np
>> X = np.random.randint(100, size=(100, 2))
>> Y = np.random.choice([0, 1], size=(100))
>> X
   array([[11, 47],
       [23,  2],
       [91, 14],
       [65, 32],
       [81, 78],
       ....
>> Y
   array([0, 0, 0, 1, 0, 1, 0, 0, 1, 0, 1, 1, 0, 1, 1, 0, 0, 1, 0, 1, 0, 1,
       0, 0, 0, 1, 1, 0, 0, 1, 1, 1, 0, 0, 0, 1, 0, 0, 0, 1, 1, 0, 1, 0,
       1, 0, 0, 1, 1, 1, 1, 1, 0, 1, 1, 0, 1, 0, 1, 1, 1, 1, 1, 0, 0, 1,
       0, 0, 1, 1, 0, 0, 1, 1, 0, 1, 0, 1, 0, 0, 1, 0, 0, 1, 1, 1, 1, 1,
       1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 1, 1])

使用海上散射图

import seaborn as sns
sns.scatterplot(x=X[:,0], y=X[:,1], hue=Y)

输出sns散射图

​

​