cate = [path + x for x in os.listdir(path) if os.path.isdir(path + x)]
>>> a=[x for x in range(10)]
>>> a
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
>>> a=[x for x in range(10) if x%2==0]
>>> a
[0, 2, 4, 6, 8]
第一句代码是用来遍历文件夹的 创建层级列表cate,用于对数据存放目录下面的数据文件夹进行遍历,os.path.isdir用于判断文件是否是目录,然后对是目录文件的文件进行遍历。
具体实现是 先执行for循环 然后判断x是否要取 如果复合条件 那么给最一开始的x