python学习案例——中国人口统计

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()  # 展示这个图

展示结果：