数据处理_2

最新推荐文章于 2025-12-11 08:56:39 发布

原创最新推荐文章于 2025-12-11 08:56:39 发布 · 388 阅读

9 ·

CC 4.0 BY-SA版权

文章标签：

#linux #python

本文介绍了使用Python的openpyxl库从文件名中提取信息，如数字和文件名，然后处理文件夹中的图片，提取这些信息并将它们写入Excel文件的过程。

from PIL import Image
import os
import openpyxl

def extract_info_from_filename(filename):
    # 解析文件名，获取两个数字和文件名
    parts = filename.split('_')
    first_number = parts[0]
    second_number = parts[1]
    file_name = parts[2].split('.')[0]  # 去掉文件扩展名(.png)
    return first_number, second_number, file_name

def process_images_and_save_to_excel(folder_path, excel_filename):
    # 创建 Excel 工作簿和工作表
    wb = openpyxl.Workbook()
    ws = wb.active

    # 写入表头
    ws.append(['First Number', 'Second Number', 'File Name'])

    # 遍历文件夹中的图片文件
    for filename in os.listdir(folder_path):
        if filename.endswith('.txt'):
            # 获取图片文件的完整路径
            image_path = os.path.join(folder_path, filename)

            # 解析文件名，提取信息
            first_number, second_number, file_name = extract_info_from_filename(filename)

            # 将信息写入 Excel 表格
            ws.append([first_number, second_number, file_name])

    # 保存 Excel 文件
    wb.save(excel_filename)
    print(f"Excel file saved as {excel_filename}")

# 指定图片文件夹路径和要保存的 Excel 文件名
image_folder_path = r'E:\code\python\data\tmp\1'
excel_filename = 'image_info.xlsx'

# 调用函数处理图片并保存到 Excel 表格
process_images_and_save_to_excel(image_folder_path, excel_filename)