python列出文件夹下所有的目录与文件

def list_all_files(rootdir):
    """
    列出文件夹下所有的目录与文件
    :param rootdir: 根路径
    :return:
    """
    _files = []
    list = os.listdir(rootdir)  # 列出文件夹下所有的目录与文件
    for i in range(0, len(list)):
        path = os.path.join(rootdir, list[i])
        if os.path.isdir(path):
            _files.extend(list_all_files(path))
        if os.path.isfile(path):
            _files.append(path)
    return _files