前言

有時(shí)處理文件夾下的文件，這時(shí)就需要使用os模塊的listdir('path')方法。

獲取文件夾下的文件和文件夾

項(xiàng)目結(jié)構(gòu)如下，test.py是腳本文件

image.png

import os
demo_list = [fn for fn in os.listdir('.')]
print(demo_list)
#輸出
['44647-106.jpg', '44648-106.jpg', '44651-106.jpg', '44653-106.jpg', 'dir1', 'dir2', 'dir3', 'main.py', 'test.py']

只需要文件夾或者jpg圖片等，則需要過濾數(shù)據(jù)。

import os
jpg_list = [fn for fn in os.listdir('.') if fn.endswith('.jpg')]
print(jpg_list)
dir_list = [fn for fn in os.listdir('.') if fn.startswith('dir')]
print(dir_list)
#輸出
['44647-106.jpg', '44648-106.jpg', '44651-106.jpg', '44653-106.jpg']
['dir1', 'dir2', 'dir3']