如何才能让我们的程序自己找路径、找文件呢?就像自动戴口罩一样。今天我们一起来探讨下,用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)等语句可以修改为对该类型文件的具体操作。