【python】+tushare庫+判斷指定日期是否是交易日

is_holiday方法：若是交易日 返回false；若不是交易日 返回true

https://www.zhihu.com/question/35835021

 

import tushare
import datetime


class AIOldData:
    def ai_trading_day(self):
        """
功能1：判斷自然日是否是交易日（YES：返回此自然日；NO：從此自然日依次往前推至交易日 並返回）
缺點：要計算10秒左右才出結果
        :return:
        """
        date_str = "{}-{}-{}".format(self[0:4], self[4:6], self[6:8])
        y, m, d = date_str.split("-")
        my_date = datetime.date(int(y), int(m), int(d))

        # 判斷日期是否是交易日
        while True:
            if tushare.is_holiday(datetime.date.strftime(my_date, "%Y-%m-%d")):
                # 不是交易日 則此日期減一天 再次循環判斷
                my_date = my_date + datetime.timedelta(days=-1)
            else:
                # 是交易日
                break
        print("離{}最近的一個歷史交易日為：{}".format(date_str, my_date))
        return my_date


if __name__ == "__main__":
    ai_old_data = AIOldData
    ai_old_data.ai_trading_day("20191007")
    ai_old_data.ai_trading_day("20191008")

 

輸出：

離2019-10-07最近的一個歷史交易日為：2019-09-30
離2019-10-08最近的一個歷史交易日為：2019-10-08