Tushare是一款開源免費的金融資料接口包,可以用于擷取股票的曆史資料.
pip install tushare
stock1.py
# coding: utf-8
import os, sys
import matplotlib.pyplot as plt
import pandas as pd
import tushare as ts
if len(sys.argv) ==2:
code = sys.argv[1]
else:
print('usage: python stock1.py stockcode ')
sys.exit(1)
if len(code) !=6:
print('stock code length: 6')
sys.exit(2)
# help(ts.get_k_data) 了解參數
df1 = ts.get_k_data(code)
df1.index = range(1, len(df1)+1) # 将index改成從1開始
# df1.head()
df1.to_csv(code+'.csv')
# 資料基本統計量
df1['close'].describe().to_csv(code+'.tsv', sep='\t')
# 取樣 2018年以後的收盤價
df1[df1['date'] >'2018-01-01']['close'].describe().to_csv(code+'.spj', sep='\t')
# help(ts.get_hist_data) 了解參數
dh = ts.get_hist_data(code)
df = dh.sort_values(by='date')
df.to_csv(code+'.txt')
運作 python stock1.py 600030