首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >在DataFrame中获取前一个工作日

在DataFrame中获取前一个工作日
EN

Stack Overflow用户
提问于 2018-10-10 22:06:27
回答 3查看 3.7K关注 0票数 13

我有一个包含两列的DataFrame,一个日期和一个类别。我想根据以下规则创建一个新的日期列:如果category是B,则值应该是最接近日期的工作日(仅来自过去或日期本身),否则是日期列本身的值。

我将工作日定义为不在周末的任何一天,也不存在于下面最小示例中定义的列表holidays中。

请考虑以下DataFrame df

代码语言:javascript
运行
复制
import datetime as dt
import pandas as pd
from IPython.display import display

holidays = [dt.datetime(2018, 10, 11)]
df = pd.DataFrame({"day": ["2018-10-10", "2018-10-11", "2018-10-12",
                       "2018-10-13", "2018-10-14", "2018-10-15"
                      ],
               "category":["A", "B", "C", "B", "C", "A"]
              }
)

df["day"] = pd.to_datetime(df.day, format="%Y-%m-%d")
display(df)

         day category
0 2018-10-10        A
1 2018-10-11        B
2 2018-10-12        C
3 2018-10-13        B
4 2018-10-14        C
5 2018-10-15        A

如何获得第三列,其值如下所示?

代码语言:javascript
运行
复制
2018-10-10
2018-10-10
2018-10-12
2018-10-12
2018-10-14
2018-10-15

我创建了一个在处理列表时查找最后一个工作日的函数,如果这有任何帮助的话。

代码语言:javascript
运行
复制
# creates a list whose elements are all days in the years 2017, 2018 and 2019
days = [dt.datetime(2017, 1 , 1) + dt.timedelta(k) for k in range(365*3)]


def lastt_bus_day(date):
    return max(
        [d for d in days if d.weekday() not in [5, 6]
                            and d not in holidays
                            and d <= date
        ]
    )

for d in df.day:
    print(last_bus_day(d))
#prints
2018-10-10 00:00:00
2018-10-10 00:00:00
2018-10-12 00:00:00
2018-10-12 00:00:00
2018-10-12 00:00:00
2018-10-15 00:00:00
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/52742118

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档