可以使用os
模塊來讀取目錄下的所有文件名。具體方法如下:
import os
def get_file_names(directory):
file_names = []
for filename in os.listdir(directory):
if os.path.isfile(os.path.join(directory, filename)):
file_names.append(filename)
return file_names
# 讀取當前目錄下的所有文件名
current_directory = os.getcwd()
files = get_file_names(current_directory)
print(files)
上述代碼中,get_file_names
函數接受一個目錄路徑作為參數,遍歷目錄下的所有文件和文件夾,通過os.path.isfile
函數判斷是否為文件,將文件名添加到file_names
列表中。最后輸出所有文件名。
如果需要遞歸地讀取目錄下的所有文件名,可以使用os.walk
函數,示例如下:
import os
def get_file_names(directory):
file_names = []
for root, dirs, files in os.walk(directory):
for filename in files:
file_names.append(filename)
return file_names
# 遞歸地讀取當前目錄下的所有文件名
current_directory = os.getcwd()
files = get_file_names(current_directory)
print(files)