Kate Li (Taiwan)的部落格

首頁

用python的matplotlib來畫全球的股票走勢圖(開源)~

作者 lace 時間 2020-04-14
all

這段時間依然是忙沒什麼值得書寫的工作和學數學方面的,這兩天學習資料分析時,順便學習用Python畫了下股票走勢圖,開源和大家分享一下(*∩_∩*)

要畫走勢圖,第一步是取得股票的數據,第二步是根據數據畫出蠟燭圖,成交量柱狀圖和移動平均線。

python的matplotlib有一個finance子模塊提供了一個獲取雅虎股票數據的api介面:quotes_historical_yahoo_ochl非常好用,直接是取得了蠟燭圖所需要的open,close,high,low數據。

而且一個介面可以取得美股,A股,日股,港股等所有股市的股票數據。

比如,如果要獲取滬深股市資訊股票名稱格式如下:

上海:股票代碼.ss

深圳:股票代碼.sz

另外還有一個庫叫panda,它的DataReader似乎更加强大,可以取得

Yahoo!Finance

Google Finance

St.Louis FED(FRED)

Kenneth French’s data library

World Bank

Google Analytics

六個地方的股票數據。但我這次沒有用panda的DataReader。

畫圖的話用

import matplotlib.pyplot as plt

plt.show()

也是超級方便。

蠟燭圖用到的是:

candles = candlestick2_ochl(ax0,opens,closes,highs,lows,width=1,colorup='blue',colordown='r')

成交量用到的是:

vc = volume_overlay(ax1,opens,closes,volumes,colorup='g',alpha=0.5,width=1)

都是matplotlib自帶的API。

我把程式碼開源了,使用的例子是:

python stock_draw.py 600028.ss cn D:\Dev\stockdb\ipagp.ttf

python stock_draw.py 002050.sz cn D:\Dev\stockdb\ipagp.ttf

python stock_draw.py 6734 jp D:\Dev\stockdb\ipagp.ttf

python stock_draw.py AAPL us D:\Dev\stockdb\ipagp.ttf

分別對應了A股,日股和美股。

效果圖:

綠盟科技(SZ:300369)

7203豐田汽車

程式碼地址:

https://gist.github.com/ithurricane/240b4aa954e09915b24697ca5f2aa1db

或者點擊原文連結。

小夥伴們有興趣繼續研究的話,可以下載

Python for Finance: Analyze Big Financial Data的電子書,更加專業的分析。

有預測風險或者走勢資料模型的小夥伴也可以私信我分享一下(☆_☆)/~~

想瞭解和關注資訊安全的小夥伴們,歡迎掃一掃關注,或者蒐索:正宗好PT