Python中的散点图

Question

我有一个尺寸为100x2的向量x，以及向量y = {1，-1}的相应二进制标签，长度为100。数据点对应于标签，例如红色为-1，对于给定数据点，黄色为1。

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

我将感谢任何帮助。

Answer 1

您可以使用seaborn（或Matplotlib）轻松执行此操作。以下是代码。

1. 我正在创建一个大小为100x2并将其称为X的随机阵列。我正在创建一个0s和1s 100x1的随机阵列，并称其为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])

使用Seaborn Scactplot

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

输出SNS STACTPLOT