python3提取指定文件夾下的所有文件名

如果你刻意練習某件事超過10000小時，那麼你就可以達到世界級。——拉里·佩奇（Google創始人）

提取指定文件夾下的所有文件名

代碼

import os

with open('./listdir.txt','a') as f:
    path = input('path:')
    try:
        list = os.listdir(path)
        for i in range(0,len(list)):
            filename = os.path.splitext(list[i])[0]
            f.write(filename + '\n')
        print('success')
    except:
        print('place input a path:')

執行

結果

封裝爲一個函數

import os

def listdir(path):
	with open('./listdir.txt','a') as f:
	    dirpath = path
	    try:
	        list = os.listdir(dirpath)
	        for i in range(0,len(list)):
	            filename = os.path.splitext(list[i])[0]
	            f.write(filename + '\n')
	        print('success')
	    except:
	        print('place input a path:')