plt.scatter画出聚类图后如何添加标签

要给 plt.scatter 画出的聚类图添加标签，可以使用 plt.annotate() 函数。这个函数可以在图形中添加文本注释，其中包括文本、箭头和标记等信息。具体地，我们可以将每个数据点的标签作为注释添加到图像中，代码如下：

import numpy as np
import matplotlib.pyplot as plt

# 聚类数据
x = np.random.randn(50, 2)
y = 2 * x + np.random.randn(50, 2) * 0.5

# 绘制散点图
plt.scatter(x[:, 0], x[:, 1], c='b', alpha=0.5)
plt.scatter(y[:, 0], y[:, 1], c='g', alpha=0.5)

# 给每个数据点添加标签
for i, (px, py) in enumerate(np.vstack((x, y))):
    plt.annotate(str(i), xy=(px, py), fontsize=12, fontweight='bold')
    
plt.show()

在这个例子中，我们使用了一个随机生成的数据集来演示如何绘制聚类图并添加标签。首先使用 `plt.scatter()` 函数绘制出两个聚类的散点图，然后使用 `plt.annotate()` 函数将每个数据点的标签添加到图像中。其中 `xy` 参数指定要注释的位置，`fontsize` 和 `fontweight` 参数控制字体大小和粗细。运行上面的代码，将得到一个带有标签的聚类图，如下所示：