Python-获取文件夹下的文件和文件夹

前言

有时处理文件夹下的文件，这时就需要使用os模块的listdir('path')方法。

获取文件夹下的文件和文件夹

项目结构如下，test.py是脚本文件

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图片等，则需要过滤数据。

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']