代码收藏家技术教程 2022-07-27

python用函数获取文件夹路径下所有文件名并生成列表

python用函数获取文件夹路径下所有文件名代码及释义如下：

### 定义文件名获取函数
def data_needed(filePath):
    import os        #引入os
    file_name = list()        #新建列表
    for i in os.listdir(filePath):        #获取filePath路径下所有文件名
        data_collect = ''.join(i)        #文件名字符串格式
        file_name.append(data_collect)        #将文件名作为列表元素填入
    print("获取filePath中文件名列表成功")        #打印获取成功提示
    return(file_name)        #返回列表
### 主函数
if __name__ == '__main__':
    filePath = "D:\\Automation\\"        #想要获取文件名的路径
    print(data_needed(filePath))        #调用文件名获取函数，并打印结果

执行结果如下所示：

获取filePath中文件名列表成功
['2022021501.txt', '2022021502.txt', '2022021505.txt', '新建 Microsoft Excel 工作表.xlsx']

进程已结束，退出代码为 0

来源：SWYR12345

Python