可以使用os.listdir()
函數來獲取指定目錄下的所有文件名。這個函數會返回一個包含目錄中所有文件和目錄名稱的列表。例如,以下代碼將獲取當前目錄(“.”)下的所有文件名:
import os
file_names = os.listdir(".")
print(file_names)
如果要獲取其他目錄下的文件名,只需將目錄路徑作為參數傳遞給os.listdir()
函數即可。例如,以下代碼將獲取/path/to/directory
目錄下的所有文件名:
import os
directory = "/path/to/directory"
file_names = os.listdir(directory)
print(file_names)
注意:os.listdir()
函數將返回目錄中所有文件和目錄的名稱,包括隱藏文件和目錄。如果只需要獲取文件名而不包括子目錄和隱藏文件,可以使用os.path.isfile()
函數來過濾結果。以下代碼將只獲取指定目錄下的文件名:
import os
directory = "/path/to/directory"
file_names = [file for file in os.listdir(directory) if os.path.isfile(file)]
print(file_names)