分享一个基于需要天数获取开始跟结束股票日期的方法

问题


每次想要获取指定天数的股票数据的时候。要么指定前几个月导致数据量太大获取太慢,要么就是数据量不够。非常麻烦。
话不多说,创建一个函数就能解决这个问题。


代码

需要的模块

import datetime
import tushare as ts
import pandas as pd

为了避免再次执行代码而产生额外的计算

可以将日期保存为csv,需要直接读取

def workday_cal(days):
    date_today = datetime.date.today()
    today = date_today.strftime('%Y%m%d')
    if not pathlib.Path('./' + today + '_code.csv').exists():
        cal_day = 0
        open_day = []
        date = ts.pro_api().trade_cal(end_date=today
                                      ).filter(
            ['is_open', 'cal_date']).sort_index(ascending=False).values
        end_day = date[0][1]
        for i in date:
            if i[0] == 1:
                cal_day = cal_day + 1
                if cal_day == days:
                    o

你可能感兴趣的:(python,开发语言)