文科生自学Python-获取指定路径下文件名

--人生不是赛场，梦想不容退场，学习编程成就更好的自己--

Python语言简洁生动，特别适合文科生学习入门IT世界，而且作为开源软件有着丰富的资源包，用几十行代码就能够做一个完整的爬虫脚本，开发效率杠杠的！短时间内即可解决工作和学习中碰到的棘手问题。(本人外语专业毕业，机缘巧合爱上编程，自学道路曲曲折折，痛并快乐！)在这里总结一下自学Python遇到的难点和重点，分享码过的代码和要点总结，希望能够给初学者一点启示和鼓励，同时愿意结交更多大神交流有助提升自己的水平。

今天分享一下如何获取指定路径下的文件名列表，例如获取某文件夹中所有EXCEL文件名列表，主要使用了OS和RE两个库。另外，Python灵魂元素众多，列表和函数两个概念不得不提，有了列表就能够进行各种遍历循环和判断，有了函数就可以方便随时调用相关功能，当然只有不断实战才会熟练的运用好这些概念。

在本地电脑某一路径下有如下文件，截图如下：

场景1：编写自定义函数切换到指定路径

代码截图如下：

文本代码如下：

import os
import re
#define a function to set work path
def Set_Work_Path(x):
    try:
        os.chdir(x) #Set work file path
        route = os.getcwd()
        print(route) #print the changed path
        return route
    except Exception:
        print("No Result")
work_path = "E:DATAFILE" #Wanted path here to change r"E:DATAFILE"
Set_Work_Path(work_path)

场景2：编写自函数提取当前路径所有特定文件名列表(文件格式为四字母)

这里主要还应用了正则表达式，正则是处理文本数据大杀器，用着就是香。

代码截图如下：

文本代码如下：

#define a function with regular expression to extract files with 4 letters in the end
def Get_All_4Letter_Files(x):
    try:
        mode = re.compile(r".*.[a-zA-Z]{4}#34;) #with 4 letter in the end files
        return re.findall(mode,x)
    except AttributeError:
        return ""
path = os.getcwd()
print(path)
files = os.listdir(path) #get all file names in the path
print(files)
List_Files_With4Letter = []
for i in range(0 ,len(files) ,1):
    List_Files_With4Letter.extend(Get_All_4Letter_Files(files[i]))
print(List_Files_With4Letter)

指定路径中所有文件名列表如下：

['Python代码总结1.txt', 'Python代码总结2.txt', 'Python代码总结3.docx.txt', 'Python代码总结3.txt', '季度报告-1.pptx', '季度报告-2.pptx', '季度报告-3.pptx', '季度报告-4.pptx', '总结1-20210724.docx', '总结1-20210724.xlsx', '总结2.docx', '总结2.xlsx', '总结3.docx', '总结3.xlsx']

通过上面代码获得结果如下，只剩下pptx,docx和xlsx这类文件格式，都是四个字母：

['季度报告-1.pptx', '季度报告-2.pptx', '季度报告-3.pptx', '季度报告-4.pptx', '总结1-20210724.docx', '总结1-20210724.xlsx', '总结2.docx', '总结2.xlsx', '总结3.docx', '总结3.xlsx']

场景3：编写自定义函数获取特定4字母格式文件名列表

代码截图如下：

文本代码如下：

#define a function to get the list of files with 4 letters in format
def Get_Dedicated_4Letter_File_List(x):
    path = os.getcwd()
    files = os.listdir(path) #print(files) #check all files name in the path
    current_list = []
    for i in range(0 ,len(files) ,1):
        try:
            if files[i][-4:] == x:
                current_list.append(files[i])
        except Exception:
            pass
    return current_list
Current_Excel_list = Get_Dedicated_4Letter_File_List("xlsx")
print(Current_Excel_list)
Current_Word_list = Get_Dedicated_4Letter_File_List("docx")
print(Current_Word_list)
Current_PPT_list = Get_Dedicated_4Letter_File_List("pptx")
print(Current_PPT_list)

通过上面代码获得结果如下，分别得到指定类型文件名列表。

['总结1-20210724.xlsx', '总结2.xlsx', '总结3.xlsx']

['总结1-20210724.docx', '总结2.docx', '总结3.docx']

['季度报告-1.pptx', '季度报告-2.pptx', '季度报告-3.pptx', '季度报告-4.pptx']

END

我为人人，人人为我！！欢迎大家关注，点赞和转发！！！