熊猫-如何根据自定义日历计算两个日期的黑白天数

Question

我的问题很简单，所以我希望有一个简单的解决方案。我想计算两个日期之间的天数，而不是使用完整的日历日或工作日或带假日日历的工作日，我想以日期列表的形式提供我自己的“日历”。假设我的日期是 ['2019-01-01', '2010-01-03', '2019-01-04', '2019-01-10']。我希望“2019-01-01”和“2019-01-03”之间的日期返回 1。“2019-01-03”和“2019-01-10”之间的日期应返回 2。谢谢！# This produces standard calendar days between-dates_list = df.indexx = dates_list[1] - dates_list[0]# This produces days according to numpy businessdaycal:cal = np.busdaycalendar()x = np.busday_count('2019-01-01', '2019-01-03', busdaycal=cal)# This works, but requires multiple steps so prob inefficient:dates_list = df.indexall_dates = pd.date_range(dates_list[0], dates_list[1])holidays = [d.date() for d in all_dates if d not in dates_list]cal = np.busdaycalendar(holidays=holidays)x = np.busday_count('2019-01-01', '2019-01-03', busdaycal=cal)

Answer 1

这是我所拥有的最好的。我尝试了@RichieV 的 pd.Series.between() 和以下方法，速度更快：

dates_list = df.index

all_dates = pd.date_range(dates_list[0], dates_list[1])

holidays = [d.date() for d in all_dates if d not in dates_list]

cal = np.busdaycalendar(holidays=holidays)

x = np.busday_count('2019-01-01', '2019-01-03', busdaycal=cal)

Answer 2

这是一种方法：

import pandas as pd

my_cal = pd.Series(

    data=1, 

    index=pd.date_range(start='2020-01-01', periods=100, freq='D'))

# set your own 'holidays' to zero here

# cumulative sum won't count your custom 'holidays'

my_cal = my_cal.cumsum()

# use like this (this could be wrapped in a function)

days_between = my_cal['2020-01-03'] - my_cal['2020-01-01']

print(days_between)