如何以Pythonic方式判断DataFrame日期是否为月末并生成新列？

先指出你原代码的几个问题：

• df['Date'].dt.is_month_end返回的是布尔值组成的Series，不需要和字符串'True'比较，直接用这个布尔Series即可。
• 不能直接用普通的if/else判断整个Series，因为条件是一组布尔值而非单一值，得用矢量化的条件判断方法。
• df['Date'].replace(day=1)用法错误，Series的replace是用来替换值的，修改日期的天应该用dt.replace(day=1)。

下面是符合你需求的Pythonic实现，全程矢量化处理，比逐行迭代高效得多：

import pandas as pd
import numpy as np
from datetime import timedelta

# 先确保Date列是datetime类型（如果还没转换的话）
df['Date'] = pd.to_datetime(df['Date'])

# 生成Date_new列
df['Date_new'] = np.where(
    df['Date'].dt.is_month_end,
    df['Date'] + timedelta(days=1),
    df['Date'].dt.replace(day=1)
)

代码说明： #

• np.where会逐个检查df['Date'].dt.is_month_end的布尔值：
  • 当值为True（即该日期是月末），就把原日期加1天；
  • 当值为False，就把原日期的天替换为1，得到当月1号。

测试你的示例：

• 日期2000/3/31是月末，处理后变为2000/4/1；
• 日期2000/3/30不是月末，处理后变为2000/3/1，完全符合需求。

内容的提问来源于stack exchange，提问作者noobcoder