使用Python获取子文件名并保存到Excel中

批量获取文件夹中的子文件名

首先你需要安装openpyxl包

可以通过下面指令实现：

pip install openpyxl

 安装好openpyxl包后就可以运行下面代码，希望对大家有帮助！

import os
from openpyxl import Workbook

def get_first_level_filenames(directory):
    # 获取目录下的所有文件名
    filenames = os.listdir(directory)
    return filenames

def save_to_excel(data):
    # 创建一个新的工作簿对象
    wb = Workbook()
    sheet = wb.active
    # 设置工作表标题
    sheet.title = '名单'
    
    # 逐行添加数据到工作表
    for filename in data:
        sheet.append([filename])
    
    # 保存工作簿的路径
    excel_file = 'F:\\提取名字.xlsx'
    wb.save(excel_file)
    print(f'File names saved to {excel_file}')  

# 需要获取的文件夹路径
directory_path = 'F:\\文件夹位置'

# 获取指定目录下第一层所有文件名
filenames = get_first_level_filenames(directory_path)
print(f'Files in directory: {filenames}')  

# 将文件名列表保存到 Excel 文件中
save_to_excel(filenames)