welcom ! Handel home

2016年5月11日 星期三

python read the direct [dir ] [file] list

[Python] 取得路徑下檔案
'ss = os.listdir(" ")      
'if  os.path.isfile(ss[0]) :
'if  os.path.isdir(ss[0]):

如何利用python取得特定路徑下的檔案名稱呢? 

其實說起來也相當簡單,os module裡已經定義好了。

先從os.listdir本身會回傳該資料夾下的所有檔案名稱,以list回傳。

再用一個 for 迴圈遍訪每個檔名,對其用os.path.isfile()判斷即可(若要取資料夾使用os.path.isdir())。

以下為範例程式碼。


import os
files = [f for f in os.listdir('.') if os.path.isfile(f)]
for f in files:
        print f
上面是個簡潔有力的寫法,他的效果等義於下列程式碼:

import os
files = []

for f in os.listdir('.'):
        if os.path.isfile(f):
                files.append(f)
for f in files:
        print f



沒有留言: