【Python量化資料】手把手教你畫出萬用精美股票圖K線圖－恩哥Python量化教室-零基礎也能學會Python

一、前言

已經學會各市場的股票資料抓下來了

【Python量化資料】用Python抓取Yahoo Finance、investing.com股價資料

量化分析的精隨在於畫圖，接下來我們來看如何畫出一張k線圖，

且之後可以利用網格佈局一直往下加入自己想要的圖，同樣手把手教到你會拉!

二、畫k線圖

這邊手把手帶大家畫出上面第一張k線圖，而第二張圖大家可以自由發揮，甚至往下加子圖~

完整程式碼

import pandas_datareader as web # 下載股票資料
import matplotlib.pyplot as plt # 基本畫圖
import mpl_finance as mpf # 畫k線
import talib # 計算技術指標
from matplotlib.gridspec import GridSpec # 畫圖網格佈局

# 下載股票資料
df = web.DataReader('2330.TW', 'yahoo')

# 計算技術指標
sma = [20, 60, 120]
for i in sma:
    df['sma'+str(i)] = talib.SMA(df['Close'], i) # talib.SMA計算簡單平均

# 轉換index
df.index = df.index.format(formatter=lambda x: x.strftime('%Y%m%d')) # 簡化x軸顯示

# 開新一個畫布 
fig = plt.figure(figsize=(15, 5)) # 寬15 長5
gs = GridSpec(2, 1) # 設立2x1的網格

# 設定網格及子圖
ax = plt.subplot(gs[0, :]) # 子圖ax在網格的[0, 全部]
ax2 = plt.subplot(gs[1, :]) # 子圖ax在網格的[1, 全部]

# k線
mpf.candlestick2_ochl(ax,
                      df['Open'],
                      df['Close'],
                      df['High'],
                      df['Low'],
                      width=0.6, alpha=0.9,
                      colorup='r', colordown='g')

# 扣抵值 # sma
sma = [20, 60, 120]
for i in sma:
    ax.plot(df.index, df['sma'+str(i)], label='sma'+ str(i)) # 均線
    x, y = df.index[-i], df['Close'].iloc[-i]*0.98
    ax.plot(x, y, 'o', label=None) # 扣抵

ax.legend(loc=2, fontsize=15) # 顯示子圖的圖例

# x軸
leng = 130 # x軸頻率 以130筆資料顯示一個x標籤
ax.set_xticks(range(0, len(df.index), leng)) # x軸頻率
ax.set_xticklabels(df.index[::leng]) # 把index放進x軸
ax.set_xlim(0, len(df.index)) # 限制x軸範圍

# y軸
ax.set_ylim(min(df['Low']) * 0.95, max(df['High']) * 1.05) # 限制y軸範圍
ax.set_title('2330.TW', fontsize=30) # 設定子圖title fontsize:大小

# 顯示格線
ax.yaxis.grid(linestyle='-.', linewidth=0.5, color='gray', alpha=0.4) # linestyle:線樣式 linewidth:粗度 
ax.xaxis.grid(linestyle='-.', linewidth=0.5, color='gray', alpha=0.4) # color:顏色 alpha:透明度
ax2.yaxis.grid(linestyle='-.', linewidth=0.5, color='gray', alpha=0.4) # linestyle:線樣式 linewidth:粗度 
ax2.xaxis.grid(linestyle='-.', linewidth=0.5, color='gray', alpha=0.4) # color:顏色 alpha:透明度ㄑ

(1) 利用mpl_finance的candlestick2_ochl 方法放入指定參數畫出k線圖

(2) 利用gs = GridSpec(2, 1) 來控制你要有多少子圖，如何擺

(3) 利用ax = plt.subplot(gs[0, :]) 來看你的子圖要放在網格哪裡

(4) df.index.format(formatter=lambda x: x.strftime('%Y%m%d')) 來控制你的x軸顯示的資料標籤轉換一下格式