Matplotlib 绘制股票走势图

引入库

import pandas as pd
import os
import matplotlib.pyplot as plt

读取数据

path = r"data/"
# os 由路径、文件夹列表、文件列表组成的元组
file = []
for path,dirs,files in os.walk(path):
    for i in files:
        fl = os.path.join(path,i)
        file.append(fl)

data = []
# 不同数据的名字
data_lable = ['Apple','Amazon','Facebook','Google','Microsoft']
for i in range(len(file)):
    dt = pd.read_csv(file[i])
    data.append(dt)
data[0]

绘图

# 创建画布，dpi表示像素
plt.figure(figsize=(16,8),dpi=300)  
# 画图
for i in range(len(data)):
    plt.plot('date', 'close', data=data[i], label=data_lable[i])

# Specify ticks for x- and y-axis
# 为了重新设置x坐标的密度，全部输出太多了，显示不了
data_date = data[0].date[::60] 
plt.xticks(data_date,data_date,rotation = 70)# 设置x周密度
# Add title and labels
plt.ylabel('Closing Price')
plt.title("Stock trend")
# Add grid
plt.grid('-')
# Add legend
plt.legend()
# Show plot
plt.show()