如何展开DataFrame以显示起止日期之间的所有日期

原始数据 #

实现步骤 #

1. 转换日期类型：将start_date和end_date转为datetime格式，方便生成连续日期序列。
2. 生成日期序列：对每一行生成从起始日期到结束日期的完整日期范围。
3. 拆分展开行：通过explode方法将日期序列拆分为单独的行，保留原行的起止日期信息。

完整代码 #

import pandas as pd

# 构造原始DataFrame
data = {
    'start_date': ['1-5-2023', '2-3-2023'],
    'end_date': ['1-10-2023', '2-10-2023']
}
df = pd.DataFrame(data)

# 转换为datetime类型
df['start_date'] = pd.to_datetime(df['start_date'], format='%m-%d-%Y')
df['end_date'] = pd.to_datetime(df['end_date'], format='%m-%d-%Y')

# 生成日期序列并展开
df['current_date'] = df.apply(lambda row: pd.date_range(row['start_date'], row['end_date']), axis=1)
df = df.explode('current_date')

# 转回原日期格式（可选操作）
df['start_date'] = df['start_date'].dt.strftime('%m-%d-%Y')
df['end_date'] = df['end_date'].dt.strftime('%m-%d-%Y')
df['current_date'] = df['current_date'].dt.strftime('%m-%d-%Y')

print(df)

运行结果 #

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