基于macd、kdj、ma技术指标分析股票多空方向——应用开发2 获取股票历史数据

这里选用tushare平台获取股票历史数据

所用包

tushare 、pandas、datetime

import tushare as ts
import pandas as pd
import datetime

获取股票历史数据

初始化tushare

未有tushare账户的亲可以到这里注册：https://tushare.pro/register?reg=486428 

#初始化你的tushare权限
token='你的token'
ts.set_token(token)
pro=ts.pro_api()

获取股票 000001.SZ 从2021-1-17到2022-1-17过去365天的数据

stock_df= pro.daily(ts_code='000001.SZ', start_date='20210117',end_date='20220117')
stock_df

输出数据如下

数据含有该股票的交易日、开盘价、最高价、最低价、收盘价、成交量等常用数据。由于我们需要获取不同股票最近一年的数据，所以对pro.daily的参数ts_code、start_date、end_date提供三个变量。分别是ts_code = stocknum、start_date=startday、end_date=today，这里需要用到datetime插件对today与startday进行初始化，代码如下。

#获取股票代码
stocknum='000001.SZ'

#获取今天日期
today = datetime.datetime.today()

#计算从今天起过去365天的日期
startday=today+datetime.timedelta(days=-365)

#使日期格式以年月日形式显示，如20220117
today = today.strftime('%Y%m%d')
startday =startday.strftime('%Y%m%d')

stock_df= pro.daily(ts_code=stocknum, start_date=startday,end_date=today)

stock_df

现在能获得动态股票最近一年的数据，但这些数据还不能直接使用需要根据需求进行优化。根据后续要求对列trade_date的数据变成时间格式并作为主索引、对列vol改名为volume、将时间从远到近排列，对此操作的代码如下。

#获取股票代码
stocknum='000001.SZ'

#获取今天日期
today = datetime.datetime.today()

#计算从今天起过去365天的日期
startday=today+datetime.timedelta(days=-365)

#使日期格式以年月日形式显示，如20220117
today = today.strftime('%Y%m%d')
startday =startday.strftime('%Y%m%d')

#获取股票历史数据
stock_df= pro.daily(ts_code=stocknum, start_date=startday,end_date=today)

#把列trade_date数据转换为时间格式
stock_df['trade_date'] = pd.to_datetime(stock_df['trade_date'])
#把trade_date作为主索引
stock_df.set_index('trade_date',inplace=True)
#把列vol改名为volume
stock_df=stock_df.rename(columns={'vol':'volume'})
#数据倒序显示
stock_df=stock_df.iloc[::-1]

stock_df

输出数据如下

由于我们的股票代码是从EXCEL表格读取的，因此我们需要导入外部表格数据传值get_stocknum给stocknum，代码如下。

#读取股票代码
get_stocknum =pd.read_excel('股票代码.xlsx')

最后把代码封装在get_stock函数里，实现stock_df=get_stock（get_stocknum.iloc[0,0]）

完整代码如下

import tushare as ts
import pandas as pd
import datetime

token='你的token'
ts.set_token(token)
pro=ts.pro_api()

#定义获取股票历史数据函数
def get_stock(num):
    stocknum=num
    today = datetime.datetime.today()
    startday=today+datetime.timedelta(days=-365)
    today = today.strftime('%Y%m%d')
    startday =startday.strftime('%Y%m%d')

    stock_df= pro.daily(ts_code=stocknum, start_date=startday,end_date=today)

    stock_df['trade_date'] = pd.to_datetime(stock_df['trade_date'])
    stock_df.set_index('trade_date',inplace=True)
    stock_df=stock_df.rename(columns={'vol':'volume'})
    stock_df=stock_df.iloc[::-1]
    return stock_df


get_stocknum =pd.read_excel('股票代码.xlsx')
#get_stocknum.iloc[0,0]值为 000001.SZ
stock_df = get_stock(get_stocknum.iloc[0,0])

stock_df