[教程]轻松掌握Python：如何快速获取文件名及文件扩展名

在Python中，获取文件名及文件扩展名是常见的操作，尤其是在文件处理和数据处理过程中。下面将详细介绍如何使用Python快速获取文件名和文件扩展名。1. 获取文件名要获取文件名，可以使用Python...

在Python中，获取文件名及文件扩展名是常见的操作，尤其是在文件处理和数据处理过程中。下面将详细介绍如何使用Python快速获取文件名和文件扩展名。

1. 获取文件名

要获取文件名，可以使用Python内置的os.path模块中的basename函数。这个函数可以从完整的文件路径中提取出文件名。

import os
# 假设有一个完整的文件路径
file_path = '/home/user/documents/report.txt'
# 使用os.path.basename获取文件名
file_name = os.path.basename(file_path)
print(file_name) # 输出: report.txt

2. 获取文件扩展名

获取文件扩展名通常需要从文件名中提取最后点（.）之后的部分。以下是一个函数，用于获取文件扩展名：

def get_file_extension(file_name): # 分割文件名和扩展名 name, ext = os.path.splitext(file_name) return ext
# 使用函数获取文件扩展名
file_extension = get_file_extension(file_name)
print(file_extension) # 输出: .txt

3. 结合使用

如果你需要同时获取文件名和扩展名，可以将上述两个步骤结合起来：

# 获取文件名和扩展名
file_name, file_extension = os.path.splitext(file_path)
print("File Name:", file_name) # 输出: report
print("File Extension:", file_extension) # 输出: .txt

4. 注意事项

• 确保文件路径是正确的，否则os.path.basename可能会返回一个不包含文件名的路径。
• 如果文件名中包含点（.），并且不是扩展名的一部分，os.path.splitext可能不会正确地分割文件名和扩展名。

5. 示例

假设你有一个包含多个文件的目录，并且想要获取每个文件的文件名和扩展名，以下是一个示例代码：

import os
# 假设这是目录路径
directory_path = '/home/user/documents'
# 获取目录中所有文件的列表
files = os.listdir(directory_path)
# 遍历文件列表，获取每个文件的文件名和扩展名
for file in files: file_path = os.path.join(directory_path, file) if os.path.isfile(file_path): # 确保是文件而不是目录 file_name, file_extension = os.path.splitext(file) print(f"File: {file_path}") print(f"File Name: {file_name}") print(f"File Extension: {file_extension}") print("---")

通过以上步骤，你可以轻松地在Python中获取文件名和文件扩展名。这些操作在文件处理和数据处理中非常有用，希望这篇文章能帮助你更好地掌握这些技巧。