如何才能让我们的程序自己找路径、找文件呢?

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

你可能感兴趣的:(如何才能让我们的程序自己找路径、找文件呢?)