分散函数如何在python中工作以绘制图形

Question

scatter 函数在这里是如何工作的？我想知道是什么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]X对表单执行索引[row, column]，其中的:意思是“一切”。因此，组合表达式的意思是“从索引为 0 的列（即第一列）中从 X 中取出每一行”。类似地，X[:, 1]从第二列中的 X 中获取每一行。这些可以一起作为散点图的 x 和 y 坐标。传递c=Y告诉函数您希望根据 的相应值对点进行着色Y。因此，所有带有 的点Y=0都将是一种颜色，而所有带有 的点Y=1都将是另一种颜色。