用tushare的trade_cal接口获取股票交易日信息,并根据pretrade_date列写一个获取下一个交易日的接口,返回str类型
时间: 2023-12-03 19:47:23 浏览: 212
c#中利用Tu Share获取股票交易信息
以下是获取下一个交易日的代码实现:
```python
import tushare as ts
# 获取股票交易日信息
cal = ts.trade_cal()
def get_next_trade_date(date):
"""
获取下一个交易日
:param date: str,日期,格式为'YYYYMMDD'
:return: str,下一个交易日,格式为'YYYYMMDD'
"""
# 将日期格式转换为datetime类型
dt = datetime.datetime.strptime(date, '%Y%m%d')
# 判断当前日期是否为交易日
while True:
if cal[cal['calendarDate']==dt.strftime('%Y-%m-%d')]['isOpen'].values[0] == 1:
break
else:
# 如果是非交易日,则继续往后推一天
dt += datetime.timedelta(days=1)
# 继续往后推一天,直到找到下一个交易日
while True:
dt += datetime.timedelta(days=1)
if cal[cal['calendarDate']==dt.strftime('%Y-%m-%d')]['isOpen'].values[0] == 1:
return dt.strftime('%Y%m%d')
```
使用示例:
```python
date = '20210701'
next_date = get_next_trade_date(date)
print(next_date) # 输出:'20210702'
```
阅读全文