python - python中的散点函数如何绘制图形

Question

这里的分散函数是如何工作的？我想知道下面的 c=Y 是什么意思，以及 X[:,0] 和 X[:,1] 是什么意思代码。

#make_blob is data set

X, Y = make_blobs(n_samples=500, centers=2, random_state=0, cluster_std=0.40) 
plt.scatter(X[:, 0], X[:, 1], c=Y, s=50, cmap='spring')
plt.show() 

Answer 1

假设make_blobs引用sklearn.datasets.make_blobs:

X 表示特征数据集，Y 表示相应的标签(目标)。因此，X 中的每一列都代表一个功能。

X[:, 0] 以 [row, column] 的形式对 X 执行索引，其中 : 表示“一切”。因此，组合表达式的意思是“从索引为 0 的列(即第一列)中的 X 中获取每一行”。

类似地，X[:, 1] 获取第二列中 X 的每一行。

这些可以一起作为散点图的 x 和 y 坐标。

传递c=Y告诉函数您希望根据Y的相应值对点进行着色。因此，所有具有 Y=0 的点都将具有一种颜色，所有具有 Y=1 的点都将具有另一种颜色。