天天看點

python tushare 讀取股票資料并存盤

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