plt.scatter(centroids[:, 0],
时间: 2023-12-04 17:39:03 浏览: 68
机器学习算法之使用Python实现KMeans算法.zip
据引用和引用[2]中的内容,可以得知plt.scatter()函数用于绘制散点图,其中centroids[:, 0]表示散点图中点的x坐标,而y坐标需要根据具体情况进行填写。如果是绘制数据集中的点,则可以使用data[:,1]表示y坐标,如果是绘制聚类中心点,则需要根据具体情况填写y坐标的值。
示例代码如下:
```python
import numpy as np
import matplotlib.pyplot as plt
# 随机生成100个点的数据集
data = np.random.randint(low=1, high=30, size=(100, 2))
# 随机生成3个聚类中心点
centroids = np.random.randint(low=1, high=30, size=(3, 2))
# 绘制数据集中的点
plt.scatter(data[:, 0], data[:, 1], s=50, c='b', alpha=0.5)
# 绘制聚类中心点
plt.scatter(centroids[:, 0], [10, 20, 30], s=200, c='r', marker='*', alpha=0.8)
plt.title('scatter diagram', fontsize=20, c='r')
plt.show()
```
在上述代码中,我们随机生成了100个点的数据集和3个聚类中心点,然后使用plt.scatter()函数绘制了数据集中的点和聚类中心点。其中,centroids[:, 0]表示聚类中心点的x坐标,而y坐标分别为10、20、30。
阅读全文