Matplotlib中折线图的绘画

一、查看数据

import pandas as pd
# 数据集是美国每个月的失业率的一个情况，DATE列为时间数据、VAlUE为失业率的值。
# 查看数据集信息
unrate = pd.read_csv("UNRATE.csv")
# 在做折线图之前，应当将时间类型的数据转化为时间戳类型。
unrate['DATE'] = pd.to_datetime(unrate['DATE'])
print(unrate.head())

DATE  VALUE
0 1948-01-01    3.4
1 1948-02-01    3.8
2 1948-03-01    4.0
3 1948-04-01    3.9
4 1948-05-01    3.5

二、在一张图上画一条折线图

import matplotlib.pyplot as plt
data = unrate[0:10]
# 利用plot函数画折线图，第一个参数为时间数据，第二个参数为值的数据。
plt.plot(data['DATE'],data['VALUE'])
# 利用xticks函数偏转横轴时间，防止出现重叠的情况，参数为偏转的角度。
plt.xticks(rotation=45)
# 横轴标签
plt.xlabel("Month")
# 纵轴标签
plt.ylabel("Unemployment Rate")
# 标题
plt.title('Monthly Unemployment Trends, 1948')
plt.show()

三、在一张图上画两条颜色不同的折线图

# 在一张图上画两条颜色不同的折线
# 新建一列数据储存月份
unrate['Month'] = unrate['DATE'].dt.month
# 利用参数c设置折线颜色
plt.plot(unrate[0:12]['Month'],unrate[0:12]['VALUE'], c='red')
plt.plot(unrate[12:24]['Month'],unrate[12:24]['VALUE'], c='blue')
plt.show()

四、在一张图上画多条不同颜色的折线

# 在一张图上画多条不同颜色的折线
# 用figure方法来控制画板大小
plt.figure(figsize=(10,6))
colors = ['red', 'blue', 'green', 'orange', 'black']
for i in range(5):
    start_index = i*12
    end_index = (i+1)*12
    subset = unrate[start_index:end_index]
    label = str(1948 + i)
    plt.plot(subset['Month'], subset['VALUE'], c=colors[i], label=label)
# 控制年份标签的位置
plt.legend(loc='best')
plt.xlabel('Month, Integer')
plt.ylabel('Unemployment Rate, Percent')
plt.title('Monthly Unemployment Trends, 1948-1952')
plt.show()

五、数据集地址

https://pan.baidu.com/s/1MbMZG-laaMax-Z4n4LIhgw