matplotlib入门

以下题为例:

假设同学1在30岁的时候,统计出来了从11岁到30岁每年
交的女朋友数量如列表y1,请绘制出该数据的折线图,以便分析每年
交女朋友的数量走势
同学2每年交女朋友数量走势如y2,绘制折线图
要求y轴表示个数,x轴表示岁数,比如11岁,12岁

from matplotlib import pyplot as plt
from matplotlib import font_manager
import random

y1 = [1, 0, 1, 1, 2, 4, 3, 2, 3, 4, 4, 5, 6, 5, 4, 3, 3, 1, 1, 1]
y2 = [random.randint(0, 3) for i in range(20)]
x = range(20)

# matplotlib不支持中文字体,可以通过以下方法调用本地字体,fname后添加字体路径
# linux中可以通过命令 fc-list 查看所有字体, fc-list :lang=zh 查看所有中文字体
my_font = font_manager.FontProperties(fname="C:\WINDOWS\FONTS\SIMKAI.TTF")

# 构造所有x轴标签
_xticks = ["{}岁".format(i) for i in range(11, 31)]

# 设置整图,figsize设置大小,dpi设置清晰度
plt.figure(figsize=(20, 8), dpi=80)

# 绘制折线图,添加颜色,标签
plt.plot(x, y1, color="orange", label="同学1")
plt.plot(x, y2, color="cyan", label="同学2")

# 添加背景网格,alpha设置透明度
plt.grid(alpha=0.3)

# 添加图例.只有图例的字体属性需要用prop设置,其余全部使用fontproperties设置
# loc设置图例位置,具体可查看源码
plt.legend(prop=my_font, loc="upper left")

# x轴刻度可以自行设置,需要量度与刻度一一对应. rotation设置旋转
plt.xticks(list(x)[::2], _xticks[::2], rotation=45, fontproperties=my_font)

# 一下分别为x,y,标题文字
plt.xlabel("年龄 单位(岁)", fontproperties=my_font)
plt.ylabel("女票个数", fontproperties=my_font)
plt.title("11-30岁每年女票个数", fontproperties=my_font)

# 保存图片,可以设置为.svg格式
plt.savefig('./1.png')

# 展示图片
plt.show()