Matplotlib学习--绘制散点图

"""
绘制简单Scatter plot
"""
import matplotlib.pyplot as plot
from pylab import mpl
import random

# 设置图表显示中文
mpl.rcParams["font.sans-serif"] = ["SimHei"]
mpl.rcParams["axes.unicode_minus"] = False  # 设置正确显示符号

plot.style.use('ggplot')

x_values = [random.randint(1,1000) for x in range(1,100)]
y_values = [x_values[i]**2 for i in range(0,99)]
fig, ax = plot.subplots()
# 色彩映射(c=y_values, cmap=plot.cm.Blues)，这里将参数c设置成了一个y值列表，并使用参数cmap告诉pyplot使用哪个颜色映射
ax.scatter(x_values, y_values, c=y_values, cmap=plot.cm.Blues, s=10)
# 设置标题与x、y轴
ax.set_title("平方数", fontsize=24)
ax.set_xlabel("值", fontsize=14)
ax.set_ylabel("值的平方", fontsize=14)
#设置刻度标记大小
ax.tick_params(axis='both', which='major',labelsize=14)
#保存图表
plot.savefig('scatter_plot.png', bbox_inches='tight')
plot.show()

运行结果：

色彩映射是一系列颜色，从起始颜色渐变到结束颜色。
更多色彩映射参考官网