python统计文件名

 

import os
import csv

def process_files(folder_path, output_csv):
    # 准备数据列表
    data = []
    serial=1
    # 遍历文件夹中的所有文件
    for filename in os.listdir(folder_path):
        print(filename)
        if filename.endswith('.pdf'):
            # 移除.pdf后缀
            name_without_ext = filename[:-4]
            
            # 按"-"分割
            parts = name_without_ext.split('-')
            print(parts)
            # 确保分割后有3部分
            if len(parts) == 2:
                # 提取编号、名称和价格
                #serial = parts[0].strip()
                name = parts[0].strip()
                price = parts[1].strip()
                
                # 添加到数据列表
                data.append([serial, name, price])
                serial=serial+1
            else:
                print("+++++++++")
        else:
            print("============")
    # 写入CSV文件
    with open(output_csv, 'w', newline='', encoding='utf-8') as csvfile:
        writer = csv.writer(csvfile)
        # 写入表头
        writer.writerow(['编号', '名称', '价格'])
        # 写入数据
        writer.writerows(data)
    
    print(f"处理完成，结果已保存到 {output_csv}")

# 使用示例
folder_path = 'D:\\发票\\2025-05-15'  # 替换为实际文件夹路径
output_csv = 'D:\\发票\\2025-05-15\\2025051502.csv'      # 输出CSV文件名
process_files(folder_path, output_csv)

 

 

#######################