以下是一个使用matplotlib库绘制按y值分组着色的分类散点图的示例代码:
import matplotlib.pyplot as plt
import numpy as np
# 生成示例数据
np.random.seed(0)
x = np.random.randn(100)
y = np.random.randn(100)
colors = np.random.randint(0, 3, 100)
# 创建散点图
plt.scatter(x, y, c=colors, cmap='viridis')
# 添加颜色图例
cbar = plt.colorbar()
cbar.set_ticks([0, 1, 2])
cbar.set_ticklabels(['Group 0', 'Group 1', 'Group 2'])
# 添加标题和坐标轴标签
plt.title('Scatter Plot with Colored Groups')
plt.xlabel('X')
plt.ylabel('Y')
# 显示图形
plt.show()
此代码使用numpy生成随机的x和y值,以及随机的颜色值。然后使用plt.scatter函数创建散点图,并通过c参数指定颜色值。cmap参数用于选择颜色映射,这里选择了'viridis'颜色映射。接下来,使用plt.colorbar()函数添加颜色图例,并设置标签和刻度。最后,通过plt.title、plt.xlabel和plt.ylabel函数添加标题和坐标轴标签。最后一行的plt.show()用于显示图形。
上一篇:any在Python中不打印内容