在 matplotlib 中设置颜色条范围

在数据可视化中，颜色条是一种常用的展示方式，用于展示数据的数值大小。在 matplotlib 中，可以通过设置颜色条范围来调整颜色条的显示效果。本文将介绍如何在 matplotlib 中设置颜色条范围。

语法

设置颜色条范围有多种方式，其中最常见的两种方式是通过 matplotlib.colors.Normalize 类和 plt.colorbar 函数。

通过 Normalize 类设置

import matplotlib.pyplot as plt
from matplotlib.colors import Normalize

# 定义数据
data = [1, 2, 3, 4, 5]

# 定义颜色和范围
colors = ['r', 'g', 'b', 'c', 'm']
cm = plt.cm.colors.ListedColormap(colors)
norm = Normalize(vmin=1, vmax=5)

#画图
plt.scatter(range(len(data)), data, c=data, cmap=cm, norm=norm)

#添加颜色条
plt.colorbar()
plt.show()

通过 plt.colorbar 函数设置

import matplotlib.pyplot as plt

# 定义数据
data = [1, 2, 3, 4, 5]

# 画图
plt.scatter(range(len(data)), data, c=data, cmap='jet')

# 添加颜色条
plt.colorbar(ticks=[1, 3, 5])

plt.show()

参数介绍

常用参数

• vmin：颜色条的最小值
• vmax：颜色条的最大值
• ticks：颜色条的刻度

总结

本文介绍了在 matplotlib 中设置颜色条范围的两种方式，并介绍了常用参数的作用。我们可以根据实际需求选择不同的方式进行设置，以达到最佳的数据可视化效果。