Python获取文件夹下文件名和文件类型
在Python中，我们经常需要遍历文件夹并获取文件名和文件类型的信息。这在很多应用中都是非常常见的需求，比如文件管理、数据处理等。本文将介绍如何通过使用Python来获取文件夹下的文件名和文件类型，并给出相关的示例代码。
方法一：使用os模块
Python的os模块提供了一组函数来与操作系统进行交互，并提供了获取文件和文件夹相关信息的功能。我们可以使用os.listdir()函数来获取文件夹中的所有文件和文件夹的名称，然后使用os.path.splitext()函数来获取文件的扩展名。
下面是一个示例代码：
import os
def get_files(folder_path):
file_list = []
for file in os.listdir(folder_path):
file_path = os.path.join(folder_path, file)
if os.path.isfile(file_path):
file_name, file_ext = os.path.splitext(file)
file_list.append((file_name, file_ext))
return file_list
folder_path = '/path/to/folder'
files = get_files(folder_path)
for file in files:
print(f"文件名：{file[0]}，文件类型：{file[1]}")
以上代码首先定义了一个名为get_files()的函数，它接受一个文件夹路径作为参数，并返回一个包含文件名和文件类型的元组列表。然后，我们使用os.listdir()函数遍历文件夹中的所有文件和文件夹，并使用os.path.isfile()函数判断是否为文件。对于文件，我们使用os.path.splitext()函数来获取文件的扩展名，并将文件名和扩展名保存在一个元组中。最后，我们打印出文件名和文件类型的信息。
方法二：使用glob模块
除了使用os模块以外，我们还可以使用glob模块来获取文件夹下的文件名和文件类型。glob模块提供了一个函数glob.glob()，它接受一个匹配模式作为参数，并返回所有匹配的文件路径。
下面是一个示例代码：
import glob
def get_files(folder_path):
file_list = []
for file_path in glob.glob(folder_path + '/*'):
if os.path.isfile(file_path):
file_name, file_ext = os.path.splitext(os.path.basename(file_path))
file_list.append((file_name, file_ext))
return file_list
folder_path = '/path/to/folder'
files = get_files(folder_path)
for file in files:
print(f"文件名：{file[0]}，文件类型：{file[1]}")
以上代码首先定义了一个名为get_files()的函数，它接受一个文件夹路径作为参数，并返回一个包含文件名和文件类型的元组列表。然后，我们使用glob.glob()函数来获取文件夹下所有的文件路径，并使用os.path.isfile()函数判断是否为文件。对于文件，我们使用os.path.basename()函数获取文件名，并使用os.path.splitext()函数来获取文件的扩展名。最后，我们打印出文件名和文件类型的信息。
总结
通过使用Python的os模块或glob模块，我们可以很方便地获取文件夹下的文件名和文件类型。无论是对于文件管理还是数据处理，这些功能都是非常实用的。希望本文对你在使用Python获取文件夹下文件名和文件类型时有所帮助。
甘特图
下面是一个使用mermaid语法标识的甘特图，展示了上述两种方法的执行时间。
gantt
dateFormat YYYY-MM-DD
title 获取文件名和文件类型的执行时间
section 方法一：使用os模块
方法一开始 :2021-01-01, 1d
方法一结束 :2021-01-02, 1d
section 方法二：使用glob模块
方法二开始 :2021-01-02, 1d
方法二结束 :2021-01-03, 1d
以上甘特图展示了方法一和方法二的执行时间，可以看出两种方法的执行时间是相近的。
以上就是使用Python获取文件夹下文件名和文件类型的方法。希望本文对你有所帮助！
