提前準備工作:
1, 安裝pandas 子產品和 tushare子產品
打開cmd,直接安裝,注意權限
![](https://img.laitimes.com/img/__Qf2AjLwojIjJCLyojI0JCLicmbw5yN5EzNjhzM5EzMjZ2NjVDN5M2NiFDNjZDMhhTMwYTYl9CX0JXZ252bj91Ztl2Lc52YucWbp5GZzNmLn9Gbi1yZtl2Lc9CX6MHc0RHaiojIsJye.png)
2, 注冊Tushare社群,并擷取’my token’
Link: https://waditu.com/register , 成功後到個人中心擷取, 注意積分達到120才有權限擷取資料。
上述工作完成之後開始撸代碼,隻需簡單幾行即可擷取:
Print結果如下:
儲存的csv檔案内容如下:
生成簡單的K線圖:
#generate to pic
import mplfinance as mpf
import numpy as np
from matplotlib import pyplot as plt
fig = plt.figure(figsize=(12,8))
ax = fig.add_subplot(111)
df_ma = pd.DataFrame()
original_data = pd.read_csv(r'C:\Users\002475.SZ.csv')
print(original_data)
days = 511
opens = original_data['open'][1:days + 1]
closes = original_data['close'][1:days + 1]
highs = original_data['high'][1:days + 1]
lows = original_data['low'][1:days + 1]
data_index = original_data['trade_date'][1:days + 1]
ohlc = list(zip(np.arange(0,len(opens)), opens, closes, highs, lows))
mpf.candlestick2_ochl(ax, opens, closes, highs, lows, width=0.6, colorup='r', colordown='g')
ax.set_xlim(0,days + 1)
ax.set_xticks(np.arange(0,days + 1,20))
ax.set_xlabel('Date', fontsize = 15)
ax.set_ylabel('Price', fontsize = 15)
ax.set_title('002475')
plt.show()
123456789101112131415161718192021222324
隻是簡單的擷取資料,沒有涉及相關的算法和實作方式,同時歡迎各位大佬提點。先行謝過!
PS:如有需要Python學習資料的小夥伴可以加點選下方連結自行擷取
python免費學習資料以及群交流解答點選即可加入