python datetimeとjpholidayを組み合わせて、平日か土日祝日かを判定したい

実現したいこと

平日か土日祝日かを判定する関数を適応したときにエラーを出ないようにしたい

実施コード

python
1def isBizDay(DATE):
2    Date = datetime.date(int(DATE[0:4]), int(DATE[4:6]), int(DATE[6:8]))
3    if Date.weekday() >= 5 or jpholiday.is_holiday(Date):
4        return 0
5    else:
6        return 1
7
8#平日か休日かを判定する新たなカラムを作成
9df['holiday'] = df['sale_date'].apply(isBizDay)

使用データフレーム

# df_1
   sale_date     ID 
0 20170630       1
1 20170701       2
2 20170702       3
3 20170703       4 
4 20170704       5

Name: sale_date, dtype: object

###　エラー内容

4036             else:
   4037                 values = self.astype(object).values
-> 4038                 mapped = lib.map_infer(values, f, convert=convert_dtype)
   4039 
   4040         if len(mapped) and isinstance(mapped[0], Series):

pandas\_libs\lib.pyx in pandas._libs.lib.map_infer()

<ipython-input-61-e1d8d88f642c> in isBizDay(DATE)
      1 def isBizDay(DATE):
----> 2     Date = datetime.date(DATE[0:4], DATE[4:6], DATE[6:8])
      3     if Date.weekday() >= 5 or jpholiday.is_holiday(Date):
      4         return 0
      5     else:

TypeError: 'int' object is not subscriptable

ご教授いただけますよう、よろしくお願いいたします。

行動規範の内容に同意します

回答2件

ベストアンサー

sale_date列は実際にはint型ではないでしょうか。であれば提示エラーが発生します。
その場合は以下のようにstrに変換してやればよいです。

Python
1import pandas as pd
2import jpholiday
3import datetime
4
5def isBizDay(DATE):
6    DATE = str(DATE) # strに変換
7    Date = datetime.date(int(DATE[0:4]), int(DATE[4:6]), int(DATE[6:8]))
8    if Date.weekday() >= 5 or jpholiday.is_holiday(Date):
9        return 0
10    else:
11        return 1
12
13df = pd.DataFrame({'sale_date':[20170630]})
14
15#平日か休日かを判定する新たなカラムを作成
16df['holiday'] = df['sale_date'].apply(isBizDay)
17print(df)