颜色映射(colormap)是一系列颜色,它们从起始颜色渐变到结束颜色。在可视化中,颜色映射用于突出数据的规律,例如,你可能用较浅的颜色来显示较小的值,并使用较深的颜色来显示较大的值。
模块pyplot内置了一组颜色映射。要使用这些颜色映射,你需要告诉Python该如何设置数据集中的每个点的颜色。
下边演示如何根据每个点的y值来设置颜色:
#!/usr/bin/env python
# -*- coding:utf-8 -*-
import matplotlib.pyplot as plt
x_values = list(range(1, 501))
y_values = [x**2 for x in x_values]
plt.scatter(x_values, y_values, s=50, c=y_values, cmap=plt.cm.Blues, edgecolors='none')
# 设置图表标题, 并给坐标轴加上标签
plt.title("Square Numbers", fontsize=24)
plt.xlabel("Value", fontsize=14)
plt.ylabel("Square of Value", fontsize=14)
# 设置刻度标记的大小
plt.tick_params(axis='both', which='major', labelsize=14)
plt.show()
代码解析:
将参数c设置成了一个y值列表,并使用参数cmap告诉Python使用哪个颜色映射。这些代码将y值较小的点显示为浅蓝色,并将y值较大的点显示为深蓝色。
运行结果: