[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
沒有留言:
張貼留言