import numpy as np
import matplotlib as mpl
import matplotlib.pyplot as plt
mpl.rcParams['font.sans-serif'] = ['FangSong']
mpl.rcParams['axes.unicode_minus'] = False
# 打开文件
with open(r'D:\Pycharm\PythonProject\Examples\Material\csv\中国人口年度数据.csv', 'r') as file:
text = file.readlines() # 每行形成一个字符串,作为列表的一个元素
year = [y for y in text[0].strip().split(',')][1:] # 注意对第一行的大字符串通过‘,’分词,并且消除多余的空格
year.reverse() # 根据数据特征进行反向排序
mess = {} # 用一个空字典统计数据
for i in range(1,6): # 依次将各行数据加入字典,其中数据名为键,数据内容为值,方便后期调用
name = text[i].split(',')[0] # 获取数据名称
name_nb = [y for y in text[i].strip().split(',')][1:] # 获取数据的值列表
name_nb = [int(i) for i in name_nb] # 将字符串全部转化为整数
name_nb.reverse() # 与year保持一致,反向排序
mess[name] = name_nb # 创建一个键值对,在循环的条件下,创建所有的键值对
mess_keys = [i for i in mess.keys()] # 建立一个键列表,方便后期(画图)调用
# 画图
plt.figure(figsize=(12,8)) # 注意格式
colors = 'rgbyc' # 设置颜色
for i in range(5): # 用循环在一幅图中画多个图
name = mess_keys[i] # 调用一个键,画一个图
Y = mess[name] # 调用这个键对应的值
# Y中的每一个值都是字符串,如果设置标签,程序无法反映,需要变成整数
# Y = [int(i) for i in Y] # 时刻注意数据的类型,应当在前面数据处理的时候就解决
plt.plot(year, Y, color=colors[i], label=name)
# 展示后发现Y轴的标签很乱,需要手动调整一下
plt.yticks(range(10000,150000,10000))
plt.ylabel('人口数') # 给y轴加标签
plt.legend() # 展示标签
# plt.savefig(r'D:\Pycharm\PythonProject\Examples\Material\png\中国人口统计图.png')
plt.show() # 展示这个图
展示结果: