用Python自动获取任意路径下文件名称的2种方法

原创 码农  2020-02-18 10:38:09  阅读 623 次 评论 0 条

新型肺炎已在全国蔓延,现在已经出现聚集性疫情,发生本地传播的风险较高!所以,一定要对聚会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文件夹下的所有文件名称,并赋值给变量filesfiles是一个存有文件名称的列表。

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.docx2.docx两个 Word 文件。

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

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

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

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

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

本文地址:https://www.itcodeit.com/post/78.html
版权声明:本文为原创文章,版权归 码农 所有,欢迎分享本文,转载请保留出处!

发表评论


表情

还没有留言,还不快点抢沙发?