In[1]:

import os

import numpy as np

import matplotlib.pyplot as plt

os.chdir('/course/DataAnalyze /data')

plt.rcParams['font.sans-serif'] = 'WenQuanYi Zen Hei'   # 设置中文显示

feature = np.load('./populations.npz', allow_pickle=True)['feature_names']  # 加载列名

amount = np.load('./populations.npz', allow_pickle=True)['data']  # 加载数据

amount = amount[::-1,:][2:,:]

p1 = plt.figure(figsize=(9,10))  #设置画布

ax1 = p1.add_subplot(2,1,1)     #创建第一幅子图

plt.title('1996-2015年人口数据散点图与折线图')

plt.ylabel('人口数量')

plt.xticks(rotation = 45)

for i in range(1,6):            #利用for循环画图

    plt.scatter(amount[:,0][0:20],amount[:,i][0:20])

ax1.legend(feature[1:],bbox_to_anchor=(1, 1),ncol=1)  #图例放图外，以免遮挡信息

ax2 = p1.add_subplot(2,1,2)     #创建第二幅子图

plt.ylabel('人口数量')

plt.xticks(rotation = 45)

for i in range(1,6):

    plt.plot(amount[:,0][0:20],amount[:,i][0:20])

ax2.legend(feature[1:],bbox_to_anchor=(1, 1),ncol=1)

plt.show()

Out[1]:

In[2]:

# 调节rc参数方便观察

# 重新画图

p1 = plt.figure(figsize=(9,10))

ax1 = p1.add_subplot(2,1,1)

plt.title('1996-2015年人口数据散点图与折线图')

plt.ylabel('人口数量')

plt.xticks(rotation = 45)

marker=['o','D','v','*','h']     #提前设置点的形状与颜色

color=['blue','darkred','red','darkgreen','green']  #颜色深浅不同但色彩相近，方便作分类对比

for i in range(1,6):

    plt.scatter(amount[:,0][0:20],amount[:,i][0:20],marker=marker[i-1],c=color[i-1])

ax1.legend(feature[1:],bbox_to_anchor=(1, 1),ncol=1)

ax2 = p1.add_subplot(2,1,2)

plt.ylabel('人口数量')

plt.xticks(rotation = 45)

linestyle=['-','--','-.',':','-']  #提前设置线段的形状

for i in range(1,6):

    plt.plot(amount[:,0][0:20],amount[:,i][0:20],linestyle=linestyle[i-1],c=color[i-1])

ax2.legend(feature[1:],bbox_to_anchor=(1, 1),ncol=1)

plt.show()

Out[2]: