5. Matplotlib折线图绘制

版权声明：转载注明出处 https://blog.csdn.net/deephacking/article/details/82731731

import pandas as pd
import matplotlib.pyplot as plt

# https://github.com/mdengler/rsandbox/blob/master/data/FRED/UNRATE.csv
unrate = pd.read_csv("UNRATE.csv")
print(unrate.head(12))
unrate["DATE"] = pd.to_datetime(unrate["DATE"])
# 新增月份列数据
unrate["MONTH"] = unrate["DATE"].dt.month
print(unrate.head(12))
# first_twelve = unrate[0:12]
# # 画图，传入x轴和y轴数据，指定折线颜色
# plt.plot(first_twelve["MONTH"],first_twelve["VALUE"], c='red')
# # 画第2条折线图，指定折线颜色
# plt.plot(unrate[12:24]["MONTH"], unrate[12:24]["VALUE"], c="blue")
# # 更改横坐标的角度
# plt.xticks(rotation = 45)
# # x轴标签
# plt.xlabel("Month")
# # y轴标签
# plt.ylabel("Unemployment Rate")
# # 图像标题
# plt.title("Monthly Unemployment Trends,1948")
# plt.show()

# 循环画多条折线
colors = ["red", "blue", "green", "orange", "black"]
for i in range(5):
    data = unrate[12*i:12*i+12]
    labels = str(1948 + i)
    plt.plot(data["MONTH"], data["VALUE"], c=colors[i], label=labels)
# 图标放置位置
plt.legend(loc="best")
plt.show()