一、查看数据
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()