使用tushare获取某股票近十年的相关数据
tushare ID:414889
前言
最近需要一些股票的往年数据来进行简单分析,所以注册了tushre,用python接口来获取数据
一、目的
使用tushare的
pro.query('fina_indicator', ts_code=query_ts_code, start_date=start)
pro.query('cashflow', ts_code=query_ts_code, start_date=start)
pro.query('balancesheet', ts_code=query_ts_code, start_date=start)
接口获取财务指标数据,资产负债表,现金流量表,并从中取出目标数据进行聚合,形成自己的目标数据。
二、代码实现
- 导入相关库,设置接口token(在网站“个人主页-接口TOKEN”查看)
import tushare as ts
import pandas as pd
pro = ts.pro_api('token')
- 定义函数,获取某只股票多年数据
start和end为开始截止日期,query_ts_code为股票代码,返回dataframe
def multi_year_data(start, end, query_ts_code):
# 财务指标数据
df1 = pro.query('fina_indicator', ts_code=query_ts_code, start_date=start,
end_date=end, fields='ts_code,end_date,roe_yearly,roa_yearly,eps,revenue_ps,equity_yoy')
df1['end_date'] = pd.to_datetime(df1['end_date'])
mask = df1['end_date'].apply(lambda x: (x.month == 12) & (x.day == 31))
df1 = df1[mask].reset_index(inplace=False, drop=True)
# 现金流量表
df2 = pro.query('cashflow', ts_code=query_ts_code, start_date=start, end_date=end,
fields='end_date,ts_code,net_profit')
df2['end_date'] = pd.to_datetime(df2['end_date'])
mask = df2['end_date'].apply(lambda x: (x.month == 12) & (x.day == 31))
df2 = df2[mask].drop_duplicates(
subset=['ts_code', 'net_profit'], keep='first')
df2.reset_index(inplace=True, drop=True)
# 资产负债表
df3 = pro.query('balancesheet', ts_code=query_ts_code, start_date=start, end_date=end,
fields='end_date,ts_code,total_assets')
df3['end_date'] = pd.to_datetime(df3['end_date'])
mask = df3['end_date'].apply(lambda x: (x.month == 12) & (x.day == 31))
df3 = df3[mask].drop_duplicates(
subset=['ts_code', 'total_assets'], keep='first')
df3.reset_index(inplace=True, drop=True)
# 三个表拼接
df = df1.merge(df2, on=['ts_code', 'end_date'])
df = df.merge(df3, on=['ts_code', 'end_date'])
col = 'ts_code,end_date,roe_yearly,roa_yearly,total_assets,equity_yoy,eps,revenue_ps,net_profit'
df = df.reindex(columns=col.split(','))
df.rename(columns={'roe_yearly': 'roe_yearly(%)',
'roa_yearly': 'roa_yearly(%)', 'equity_yoy': 'equity_yoy(%)'}, inplace=True)
return df
获取平安银行近10年的相关数据(数据说明见tushare官网接口说明,如财务指标数据)
multi_year_data('20100101', '20201231', '000001.SZ')
近10年年报数据
多次调用这个函数即可获得多支想要的股票数据