python自动化系列:批量导出 Excel 数据到 CSV 文件

import osimport pandas as pd

def find_files_with_extension(directory, extension):    """从给定目录下查找特定扩展名的所有文件"""    return [os.path.join(root, file)            for root, dirs, files in os.walk(directory)            for file in files if file.endswith(extension)]

def export_excel_to_csv(excel_file, output_directory, sheet_name=None):    """    将Excel文件中的数据导出到CSV文件。    :param excel_file: Excel文件路径    :param output_directory: CSV文件输出目录    :param sheet_name: 指定要导出的工作表名称；如果为None，则导出所有工作表    """    # 确保输出目录存在    os.makedirs(output_directory, exist_ok=True)    # 读取Excel文件    xls = pd.read_excel(excel_file, sheet_name=sheet_name)    base_filename = os.path.splitext(os.path.basename(excel_file))[0]    if isinstance(xls, dict):  # 如果有多个工作表        for sheet, df in xls.items():            csv_filename = f"{base_filename}_{sheet}.csv"            csv_filepath = os.path.join(output_directory, csv_filename)            df.to_csv(csv_filepath, index=False, encoding='utf-8-sig')  # 使用utf-8-sig支持中文字符            print(f"已导出 '{excel_file}' 中的工作表 '{sheet}' 到 '{csv_filepath}'")    else:  # 如果只有一个工作表        csv_filename = f"{base_filename}.csv"        csv_filepath = os.path.join(output_directory, csv_filename)        xls.to_csv(csv_filepath, index=False, encoding='utf-8-sig')        print(f"已导出 '{excel_file}' 到 '{csv_filepath}'")

def main(input_directory, output_directory, sheet_name=None):    # 查找所有Excel文件    excel_files = find_files_with_extension(input_directory, '.xlsx')    # 遍历所有Excel文件并导出为CSV文件    for excel_file in excel_files:        export_excel_to_csv(excel_file, output_directory, sheet_name)

if __name__ == "__main__":    input_directory = '../../files'  # 替换为你的Excel文件所在的目录路径    output_directory = './csv_files'  # 替换为你的CSV文件输出目录路径    sheet_name = None  # 如果只想导出特定工作表，请指定其名称；否则设为None导出所有工作表    main(input_directory, output_directory, sheet_name)