Python實用技巧之如何獲取后綴名(擴展名)或文件名

更新時間：2024年12月31日 10:14:01 作者：高斯小哥

這篇文章主要介紹了在Python中獲取文件名和擴展名的幾種方法,包括使用os.path.basename()函數(shù)獲取文件名,以及使用os.path.splitext()函數(shù)獲取文件名和擴展名,文中通過代碼介紹的非常詳細,需要的朋友可以參考下

一、引言

在日常編程中，我們經(jīng)常需要處理文件，包括讀取、寫入、移動、復制等操作。而在這些操作中，獲取文件的擴展名（后綴名）或文件名是一個常見的需求。Python提供了多種方法來實現(xiàn)這一需求，本文將詳細介紹其中的幾種方法，并通過實戰(zhàn)案例來加深理解。

二、獲取文件名

在Python中，獲取文件名通常使用os.path.basename()函數(shù)。這個函數(shù)接受一個文件路徑作為參數(shù)，并返回該文件的基本名稱，即最后一個路徑分隔符后面的部分。

下面是一個簡單的示例：

import os

# 定義一個文件路徑
file_path = "/home/user/documents/example.txt"

# 使用os.path.basename()獲取文件名
file_name = os.path.basename(file_path)

print(file_name)  # 輸出: example.txt

??注意：os.path.basename()函數(shù)僅返回文件名，不包括路徑或擴展名。

三、獲取文件擴展名

在Python中，獲取文件擴展名通常使用os.path.splitext()函數(shù)。這個函數(shù)接受一個文件路徑作為參數(shù)，并返回一個元組，其中第一個元素是文件的基本名稱（不包括擴展名），第二個元素是文件的擴展名（包括點號）。

下面是一個簡單的示例：

import os

# 定義一個文件路徑
file_path = "/home/user/documents/example.txt"

# 使用os.path.splitext()獲取文件名和擴展名
file_name, file_extension = os.path.splitext(file_path)

print("文件名:", file_name)  # 輸出: 文件名: /home/user/documents/example
print("擴展名:", file_extension)  # 輸出: 擴展名: .txt

??注意：os.path.splitext()函數(shù)返回的擴展名包括點號（.）。如果你想要去掉點號，可以通過字符串切片操作來實現(xiàn)：

file_extension = file_extension[1:]  # 去掉點號
print("去掉點號的擴展名:", file_extension)  # 輸出: 去掉點號的擴展名: txt

四、實戰(zhàn)案例

現(xiàn)在，我們來看一個實戰(zhàn)案例，演示如何在實際編程中運用獲取文件名和擴展名的知識。假設我們有一個文件夾，里面包含了多個文件，我們需要遍歷這個文件夾，打印出每個文件的文件名和擴展名。

import os

# 定義一個文件夾路徑
folder_path = "/home/user/documents"

# 遍歷文件夾中的文件
for file_name in os.listdir(folder_path):
    # 拼接文件的完整路徑
    file_path = os.path.join(folder_path, file_name)
    
    # 判斷是否為文件（排除文件夾）
    if os.path.isfile(file_path):
        # 使用os.path.splitext()獲取文件名和擴展名
        file_base_name, file_ext = os.path.splitext(file_name)
        
        # 打印文件名和擴展名
        print(f"文件名: {file_base_name}, 擴展名: {file_ext[1:]}")

這個示例代碼會遍歷指定文件夾中的所有文件，并打印出每個文件的文件名和擴展名。

??注意： 我們在打印擴展名時使用了字符串切片操作來去掉點號。

五、總結(jié)

本文介紹了在Python中獲取文件名和擴展名的幾種方法，包括使用os.path.basename()函數(shù)獲取文件名，以及使用os.path.splitext()函數(shù)獲取文件名和擴展名。通過實戰(zhàn)案例的演示，我們了解了如何在實際編程中運用這些知識。希望本文能夠幫助你更好地處理文件相關的編程任務！

附：python獲取指定文件夾下所有指定后綴文件列表

import os
# 獲取指定路徑下所有指定后綴的文件
# dir 指定路徑
# ext 指定后綴，鏈表&不需要帶點 或者不指定。例子：['xml', 'java']
def GetFileFromThisRootDirV2(target_dir, target_suffix="pxy"):
    find_res = []
    target_suffix_dot = "." + target_suffix
    walk_generator = os.walk(target_dir)
    for root_path, dirs, files in walk_generator:
        if len(files) < 1:
            continue
        for file in files:
            file_name, suffix_name = os.path.splitext(file)
            if suffix_name == target_suffix_dot:
                find_res.append(os.path.join(root_path, file))
    return find_res
    
LogList = GetFileFromThisRootDirV2("D:/temp", "log")

到此這篇關于Python實用技巧之如何獲取后綴名(擴展名)或文件名的文章就介紹到這了,更多相關Python獲取后綴名或文件名內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: