Python获取指定目录下的文件夹名列表和文件名列表

一、问题描述

        在下图的目录中，我想通过Python的方式来分别获取目录列表（AI，English，Math）和文件名列表（readme.txt，读书进展.docx）。

二、实现

        代码

import os

# enter of program
if __name__ == '__main__':
    # input
    input_path = 'D:/data/pycodes/books/'

    # initialize
    dir_list = []
    file_list = []

    # process
    for root, dirs, files in os.walk(input_path):
        dir_list = dirs
        file_list = files
        break

    # output
    print(dir_list, len(dir_list))
    print(file_list, len(file_list))

        效果（程序输出）

['AI', 'English', 'Math'] 3
['readme.txt', '读书进展.docx'] 2

三、参考

        Python获取指定目录下所有文件名