想写一个能获得文件夹内所有文件名字，并写入Excel的脚本

Question

path=r"D:\各种模\选股结果"
os.listdir(path)
['ipynb_checkpoint','module.ipynb','文件夹1','文件夹2','表格2.xlsx']

Answer 1

我们也可以对os.listdir()函数的结果进行筛选，将目录排除在外，只保留文件名。

from openpyxl import Workbook
import os

#定义函数：功能为获取到的文件列表写入excel文件。
#接收文件列表作为参数，并根据使用的操作系统决定存储路径（虽然一般没有必要，但是在指定路径下的存储需要使用此功能）
def write_in(obj_lst, path = os.getcwd(), plat = os.sys.platform):
    wb = Workbook()#新建工作簿
    sht = wb.active#激活工作表
    for i in range(len(obj_lst)):
        sht[f'A{i+1}'] = obj_lst[i]#将文件名依次存储在最左边一列中
    if plat == 'darwin':#判断系统
        save_path = f'{path}/ReadMe.xlsx'
    else:
        save_path = f'{path}\\ReadMe.xlsx'
    wb.save(save_path)#保存文件
    print(f'已完成写入。\nexcel文件路径为{save_path}')#提示完成进度和文件保存位置
    return save_path

#定义函数：获取指定目录下的所有文件名
def get_obj(path):
    names = os.listdir(path)
    files = []
    for i in names:
        if '.' in i:
            files.append(i)
    return files

#定义函数：与用户交互
def inter():
    path = input('请输入要检索的目录的绝对路径\n>>[')
    write_in(get_obj(path))

#调用函数
if __name__ == '__main__':
    inter()

如果您所说的“文件”也包含目录（即文件夹）的话，请使用如下代码：

from openpyxl import Workbook
import os
def write_in(path, obj_lst, plat):
    wb = Workbook()
    sht = wb.active
    for i in range(len(obj_lst)):
        sht[f'A{i+1}'] = obj_lst[i]
    if plat == 'darwin':
        save_path = f'{path}/ReadMe.xlsx'
    else:
        save_path = f'{path}\\ReadMe.xlsx'
    wb.save(save_path)
    print(f'已完成写入。\nexcel文件路径为{save_path}')
    return save_path
def get_obj(path):
    names = os.listdir(path)
    files = []
    for i in names:
        files.append(i)
    return files
def inter():
    path = input('请输入要检索的目录的绝对路径\n>>[')
    write_in(os.getcwd(), get_obj(path), os.sys.platform)
inter()