介紹
Python語言中,使用os模塊可以對操作系統進行訪問,包括文件、文件夾等訪問。os.path模塊提供了一些函數,用於處理路徑字元串。os.path.listdir()函數用於獲取指定路徑下的所有文件和文件夾的名稱,並以列表的形式返回。
正文
一、listdir()函數的使用方法
os.path.listdir()函數的基本語法如下:
import os os.listdir(path)
其中,path是需要獲取文件名和文件夾名的路徑。如果想要獲取當前路徑下的文件和文件夾,可以使用os.getcwd()函數獲取路徑。
函數返回的結果是一個列表,包含了指定路徑下所有文件名和文件夾名。下面是一個例子:
import os current_folder_path = os.getcwd() files_and_folders = os.listdir(current_folder_path) print(files_and_folders)
如果將代碼保存該文件夾下,並執行該代碼,結果可能類似於:
['file1.jpg', 'file2.txt', 'folder1', 'folder2']
二、listdir()函數的應用場景
os.path.listdir()函數可以用於很多場景,比如:
1. 獲取指定文件夾下的所有文件名和文件夾名
可以使用os.path.listdir()函數獲取指定文件夾下的所有文件名和文件夾名,並對這些元素進行處理。
import os folder_path = 'folder1' files_and_folders = os.listdir(folder_path) print(files_and_folders)
結果可能如下:
['file1.jpg', 'file2.txt']
2. 遍歷文件夾的所有文件名和文件夾名
可以使用os.path.foreach()函數獲取指定文件夾下的所有文件名和文件夾名,並利用遞歸函數對文件夾進行遍歷。
import os def traverse_folder(folder_path): files_and_folders = os.listdir(folder_path) for file_or_folder_name in files_and_folders: file_or_folder_path = os.path.join(folder_path, file_or_folder_name) if os.path.isdir(file_or_folder_path): traverse_folder(file_or_folder_path) else: print(file_or_folder_path) current_folder_path = os.getcwd() traverse_folder(current_folder_path)
3. 根據文件名和文件夾名進行篩選
可以使用os.path.listdir()函數獲取指定文件夾下的所有文件名和文件夾名,並利用if語句篩選需要的文件名和文件夾名。
import os folder_path = 'folder1' files_and_folders = os.listdir(folder_path) for file_or_folder_name in files_and_folders: if file_or_folder_name.endswith('.txt'): print(file_or_folder_name)
結果可能如下:
['file2.txt']
小結
os.path.listdir()函數是Python中獲取文件名和文件夾名的重要函數,可以用於很多場景。我們可以結合其他函數和語句,對文件和文件夾進行遍歷、篩選等操作。在進行路徑操作時,建議使用os.path模塊提供的函數,因為這些函數會考慮跨平台的問題,能夠保證在不同的操作系統上都能正常工作。
原創文章,作者:小藍,如若轉載,請註明出處:https://www.506064.com/zh-tw/n/303239.html