以下是一个示例代码,演示如何按照第三个值给情节线着色,并添加一个颜色条:
import numpy as np
import matplotlib.pyplot as plt
# 创建一些示例数据
x = np.arange(10)
y = np.arange(10)
z = np.random.randint(0, 10, size=10)
# 创建一个颜色映射
cmap = plt.cm.get_cmap('rainbow')
# 绘制散点图,并按照第三个值着色
plt.scatter(x, y, c=z, cmap=cmap)
# 添加颜色条
colorbar = plt.colorbar()
colorbar.set_label('第三个值')
# 显示图形
plt.show()
在这个示例中,我们首先创建了一些示例数据,包括x、y坐标和z值。然后我们使用plt.scatter()
函数绘制散点图,并使用cmap
参数设置颜色映射。该参数接受一个颜色映射对象,我们使用plt.cm.get_cmap()
函数来获取一个预定义的颜色映射对象,这里使用的是'rainbow'。然后我们传入z值作为c
参数,表示按照z值来着色。最后,我们使用plt.colorbar()
函数添加一个颜色条,colorbar.set_label()
函数设置颜色条的标签。最后调用plt.show()
函数显示图形。