如何在python中绘制30个要素的分布图并按标签进行区分

import matplotlib.pyplot as plt
import numpy as np

features = np.random.randn(30)  # 生成30个随机要素数据
labels = np.random.randint(0, 3, 30)  # 生成30个随机标签，范围在0到2之间

plt.scatter(range(len(features)), features, c=labels, cmap='viridis')
plt.colorbar()  # 添加颜色标签
plt.xlabel('Index')  # 设置x轴标签
plt.ylabel('Feature Value')  # 设置y轴标签
plt.title('Distribution of Features')  # 设置图表标题
plt.show()  # 显示图表