python：mplfinance 画K线图

原创

已于 2022-05-02 20:46:23 修改 · 4.2k 阅读

标签

#python #mplfinance

收录于

于 2021-11-14 14:04:53 首次发布

本文介绍了Python的mplfinance模块替代了已弃用的mpl_finance，并展示了如何使用mplfinance绘制股票K线图，包括设置图像类型、移动平均线、显示非交易日等功能。示例代码中，通过tushare获取股票数据，然后利用mplfinance的plot函数绘制了带有自定义颜色风格的K线图和移动平均线。

python的 mpl_finance模块从2020年已经提醒弃用,2021年现在用 mplfinance模块.

安装 pip install --upgrade mplfinance

mpf_kline.py

# coding: utf-8
import os, sys
import datetime
import matplotlib.pyplot as plt
import mplfinance as mpf
import pandas as pd
import tushare as ts
##绘制K线图+移动平均线
 
if len(sys.argv) ==2:
    code = sys.argv[1]
else:
    print('usage: python mpf_kline.py stockcode ')
    sys.exit(1)

if len(code) !=6:
    print('stock code length: 6')
    sys.exit(2)
    
# help(ts.get_k_data) 了解参数
df = ts.get_k_data(code)
df = df[ df['date'] >'2021-01-01']
#print(df.tail())
if len(df) <10:
    print(" len(df) <10 ")
    sys.exit(2)

df.index = pd.to_datetime(df.date)
#print(mpf.available_styles()) 
#mpf.plot(df, style='charles', mav=(5,10,20), volume=True, title=code)
#mpf.plot(df, type='line', mav=(5,10,20), title=code)
# 绘制K线图
my_color = mpf.make_marketcolors(up='red', down='green', edge='black', wick='black', volume='cyan