如何使用Python筛选文件名包含每月最后一天日期的.xlsx文件

实现代码 #

import os
import pandas as pd
import re
from pandas.tseries.offsets import MonthEnd

# 替换为你的xlsx文件存储路径
path = "你的文件夹实际路径"

# 匹配符合命名规则的文件
files = [f for f in os.listdir(path) if re.match(r'ABC DEF [0-9]+\.xlsx', f)]
df = pd.DataFrame(files, columns=['file_name'])

# 提取日期字段：修正原有索引错误，按空格拆分后日期位于索引2的位置
df['port_date'] = df['file_name'].apply(lambda x : x.split(' ')[2].split('.')[0])
# 转换为datetime格式
df['port_date'] = pd.to_datetime(df['port_date'], format='%Y%m%d')
# 计算对应月份的最后一天
df['month_end'] = df['port_date'] + MonthEnd(0)

# 筛选得到所有月末日期对应的文件名列表
target_files = df[df['port_date'] == df['month_end']]['file_name'].tolist()

逻辑说明 #

• 修正了原有代码拆分文件名取日期的索引错误：按ABC DEF 20150501.xlsx的命名规则，按空格拆分后只有3个元素，索引从0开始计数，日期部分位于索引2的位置，原代码写的索引3会触发索引越界报错。
• 内置的MonthEnd(0)偏移量会自动处理闰年、大小月的月末日期计算，无需手动写逻辑判断每月天数，可靠性更高。
• 最终得到的target_files就是所有文件名日期为对应月份最后一天的文件列表，可直接用于后续读取处理。

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