這段時間依然是忙沒什麼值得書寫的工作和學數學方面的,這兩天學習資料分析時,順便學習用Python畫了下股票走勢圖,開源和大家分享一下(*∩_∩*)
要畫走勢圖,第一步是取得股票的數據,第二步是根據數據畫出蠟燭圖,成交量柱狀圖和移動平均線。
python的matplotlib有一個finance子模塊提供了一個獲取雅虎股票數據的api介面:quotes_historical_yahoo_ochl非常好用,直接是取得了蠟燭圖所需要的open,close,high,low數據。
而且一個介面可以取得美股,A股,日股,港股等所有股市的股票數據。
比如,如果要獲取滬深股市資訊股票名稱格式如下:
上海:股票代碼.ss
深圳:股票代碼.sz
另外還有一個庫叫panda,它的DataReader似乎更加强大,可以取得
- Yahoo!Finance
Yahoo!Finance
- Google Finance
Google Finance
- St.Louis FED(FRED)
St.Louis FED(FRED)
- Kenneth French’s data library
Kenneth French’s data library
- World Bank
World Bank
- Google Analytics
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