我画了一个PCA降维图，我想在这个图中加上图例：将疾病对应的颜色在旁边注释出来

你可以在绘制PCA降维图时，使用`matplotlib`库中的`scatter()`函数绘制散点图，并在图例中添加疾病名称和对应的颜色。以下是一个简单的示例代码：

import matplotlib.pyplot as plt
import numpy as np

# 假设有三种疾病，对应三种颜色
diseases = ['A', 'B', 'C']
colors = ['red', 'green', 'blue']

# 生成随机数据（假设有100个样本，每个样本有3个特征）
data = np.random.randn(100, 3)
# 假设每个样本有一个疾病标签（0表示A疾病，1表示B疾病，2表示C疾病）
labels = np.random.randint(0, 3, size=100)

# 绘制PCA降维图
fig, ax = plt.subplots()
for i, disease in enumerate(diseases):
    # 提取对应疾病的样本和标签
    idx = np.where(labels == i)[0]
    x = data[idx, 0]
    y = data[idx, 1]
    # 绘制散点图
    ax.scatter(x, y, c=colors[i], label=disease)

# 添加图例
ax.legend()

# 显示图像
plt.show()

运行这段代码后，你将得到一个带有图例的PCA降维图，类似于下面这张图：