让你的程序学会自己戴'口罩'！让你的程序更聪明？

新型肺炎已在全国蔓延，现在已经出现聚集性疫情，发生本地传播的风险较高！所以，一定要对聚会say "NO!"。

老老实实在家待着，为国家做贡献！对自己负责，也对他人负责！如果一定要出门，务必戴上口罩啊！切不可大意！

言归正传，我们利用宅在家里的时间好好学习，充实一下自己吧……

如何让你的程序更聪明呢？？？

我们在使用 Python 进行自动办公操作的时候，经常需要手动输入文件路径和文件名称，操作繁琐不说，关键是没有真正的实现自动化啊！

如何才能让我们的程序自己找路径、找文件呢？就像自动戴口罩一样。今天我们一起来探讨下，用Python自动获取任意路径下文件名称的 2 种方法。

★

下面给大家分享一下用 Python 自动获取任意路径下文件名称的 2 种方法。

”

方法一：os 方法

这个方法是我经常使用的，比较灵活，只需要用到os库。

import os
path = os.getcwd() # 获取程序所在文件夹路径
print('程序所在文件夹路径是：'+path)

files = os.listdir(path) # 获取指定路径下的所有文件的文件名称
for file in files:
    if file.split('.')[-1] in ['txt']:
        print('这是一个文本文件：' + file)
    elif file.split('.')[-1] in ['doc','docx']:
        print('这是一个Word文件：' + file)
    elif file.split('.')[-1] in ['xls','xlsx']:
        print('这是一个Excel文件：' + file)

1、获取路径

我们将.py文件放到待处理文件夹下，通过os.getcwd()获取文件夹路径，并赋值给变量path。

2、获取文件名称列表

通过os.listdir(path)命令，获取path文件夹下的所有文件名称，并赋值给变量files，files是一个存有文件名称的列表。

3、判断文件类型

通过 for 循环，依次取出列表中的每个文件名称。文件名称都是长这个样子的：新建文件.txt、新建文件.docx、新建文件.xls、新建文件.xlsx等。

因此，我们可以通过文件后缀判断文件类型。

这里要用到字符串操作的split()函数，以.号将文件名称分隔成列表：file.split('.')，并用[-1]取出这个列表的最后一个字符，用作 if 语句的判断条件。

我们可以将需要匹配的文件类型放到列表里，如：if file.split('.')[-1] in ['txt','doc','docx','xls','xlsx','ppt','pptx']:，这也是这种方法的优势，可以一次匹配一种或多种文件类型。

方法二：os + glob 方法

这种方法语法更简练，适用于单一文件类型的匹配。需要用到os库和glob库。

import glob
import os
files = glob.glob(os.path.join(path,'*.txt') # 匹配以'*.txt'为后缀的所有文件的全路径
for file in files:
    print(file)

核心语句就是：files = glob.glob(os.path.join(path,'*.txt')

• path：是程序所在文件夹路径，可以自动获取，也可以自己指定。

• '*.txt'：匹配 .txt 后缀的文件，也可以指定自己需要的后缀名称。

这个方法获取的files列表和上一方法获取的files列表略有不同。方法一的files列表是单个文件名称，方法二的files列表是包含文件名称的绝对路径。

例如：D:\data\文件夹下有1.docx、2.docx两个 Word 文件。

方法一获取的files列表为：['1.docx','2.docx'];

方法二获取的files列表为：['D:\\data\\1.docx','D:\\data\\2.docx']。

以上就是给大家分享的，用 Python 自动获取任意路径下文件名称的 2 种方法，这是 Python 办公自动化必备小技能！

只要将你的.py文件放在待处理文件夹下，然后加上上述代码，你的程序就可以自己找路径、找文件，不受文件位置和文件名称的限制！

文中的print('这是一个文本文件：' + file)等语句可以修改为对该类型文件的具体操作。

有兴趣的朋友可以看下我的办公自动化相关文章：

1、让程序帮你干活！批量转换任意文件夹下Word文档为PDF。

2、用 Python 处理3万多条数据，只要几秒钟……

3、收藏！Python内置的轻量级数据库竟如此好用！全网最实用sqlite3实战项目。

本文完，谢谢阅读！

★

本文首发【智能演示】微信公众号，欢迎关注！公众号后台回复【课程】领取精选视频课程。

”

---

扫描下方二维码，
回复【课程】获取精选视频课程！

觉得好看就点个在看吧……