熊猫在特定月份的两个日期之间获得几天

Question

我有一个三列的pandas数据框。 开始和结束日期以及一个月。

我想为一列添加两个日期之间一个月中的多少天。 我开始使用apply ， calendar库和一些数学来做一些事情，但是它开始变得非常复杂。 我敢打赌pandas有一个简单的解决方案，但正在努力寻找它。

输入：

import pandas as pd
df1 = pd.DataFrame(data=[['2017-01-01', '2017-06-01', '2016-01-01'],
                         ['2015-03-02', '2016-02-10', '2016-02-01'],
                         ['2011-01-02', '2018-02-10', '2016-03-01']],
                  columns=['start date', 'end date date', 'Month'])

所需输出：

   start date end date date       Month  Days in Month
0  2017-01-01    2017-06-01  2016-01-01              0
1  2015-03-02    2016-02-10  2016-02-01             10
2  2011-01-02    2018-02-10  2016-03-01             31

Answer 1

有一个解决方案：通过pd.date_range start日期和end日期之间的日期列表，然后检查与目标月份具有相同year和month日期。

def overlap(x):
    md = pd.to_datetime(x[2])
    cand =  [(ad.year, ad.month) for ad in pd.date_range(x[0], x[1])]
    return len([x for x in cand if x ==(md.year, md.month)])
df1["Days in Month"]= df1.apply(overlap, axis=1)

你会得到：

   start date end date date       Month  Days in Month
0  2017-01-01    2017-06-01  2016-01-01              0
1  2015-03-02    2016-02-10  2016-02-01             10
2  2011-01-02    2018-02-10  2016-03-01             31

Answer 2

您可以通过以下方式将单元格转换为日期时间

df = df.applymap(lambda x: pd.to_datetime(x))

然后找到具有功能的交叉日

def intersectionDaysInMonth(start, end, month):
    end_month = month.replace(month=month.month + 1)
    if month <= start <= end_month:
        return end_month - start
    if month <= end <= end_month:
        return end - month
    if start <= month < end_month <= end:
        return end_month - month

    return pd.to_timedelta(0)

然后申请

df['Days in Month'] = df.apply(lambda row: intersectionDaysInMonth(*row).days, axis=1)