最近看到一个很有意思的文章,跟大家分享文章内容:
方法一:
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)
解读:
获取路径:
将.py文件放到待处理文件夹下,通过os.getcwd()获取文件夹路径,并赋值给变量path。
获取文件名称列表:
通过os.listdir(path)命令,获取path文件夹下的所有文件名称,并赋值给变量files,files是一个存有文件名称的列表。
判断文件类型:
通过 for 循环,依次取出列表中的每个文件名称。文件名称都是长这个样子的:新建文件.txt、新建文件.docx、新建文件.xls、新建文件.xlsx等。
这里要用到字符串操作的split()函数,以.号将文件名称分隔成列表:file.split(’.’),并用[-1]取出这个列表的最后一个字符,用作 if 语句的判断条件。
我们可以将需要匹配的文件类型放到列表里,如:if file.split(’.’)[-1] in [‘txt’,‘doc’,‘docx’,‘xls’,‘xlsx’,‘ppt’,‘pptx’]:,这也是这种方法的优势,可以一次匹配一种或多种文件类型。
方法二:
适用于单一文件类型的匹配。
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)等语句可以修改为对该类型文件的具体操作。