<span style="font-family: Arial, Helvetica, sans-serif; background-color: rgb(255, 255, 255);"> 今天碰到一个坑,python的列表问题,错误往往出在你最熟悉的地方,特别是小地方,问题:我要列出目录下所有的下一级文件和子目录(只包含下一级目录而不包括目录的目录及文件这种情况),然后只获取当前子目录。。问题描述很清楚,很轻易的就写了个函数,不就是os.listdir(path)嘛,对就是这个,这个功能可以列出下一级的所有目录和文件,返回结果在一个列表中,那我们的主要工作的常规思路是不是就是把这个结果列表进行过滤?好,os也有相关的功能,os.path.isfile(filepath),就可以判断文件是不是一个文件,是文件,那就从列表删除,处理完,就是我们要的最终结果列表了,是不是这样呢?想法很完美,现实很残酷,很意外,这个搞不定的。我们用代码实验:</span>
import os def Count_SubDir(src): listall = os.listdir(src) for line in listall: filepath = os.path.join(src,line) if os.path.isfile(filepath): listall.remove(line) return listall if __name__ == "__main__": src = r'f:\src' ret = Count_SubDir(src) for line in ret: print line
输出结果:
>>> ================================ RESTART ================================ >>> 1025 1028 1029 1030 1031 1032 1033 1035 1036 1037 1038 1040 1041 1042 1043 1044 1045 1046 1049 1053 1055 2052 2070 3076 3082 Graphics NDP40-KB2894842.msp Setup.exe SetupUi.dll SetupUtility.exe sqmapi.dll UiInfo.xml >>>
看结尾。。。还依然有文件存在,我们对比下没经过处理的原结果列表:
>>> ================================ RESTART ================================ >>> 1025 1028 1029 1030 1031 1032 1033 1035 1036 1037 1038 1040 1041 1042 1043 1044 1045 1046 1049 1053 1055 2052 2070 3076 3082 DHtmlHeader.html Graphics header.bmp NDP40-KB2894842.msp ParameterInfo.xml Setup.exe SetupEngine.dll SetupUi.dll SetupUi.xsd SetupUtility.exe SplashScreen.bmp sqmapi.dll Strings.xml UiInfo.xml watermark.bmp >>>
最后的文件项目,可以对比下,是不是发现什么规律了。。。隔一个漏掉一个。。。
那为什么会出现这种情况呢?就是因为for循环和list.remove()方法的配合有问题,把流程自己走一遍,问题就暴露出来了
我们走一遍:首先for循环,每次循环选中一个line,如果这个line是文件,从list中删除,然后第二次循环。。。。问题来了,是不是漏掉了什么?删除之后,是不是还有什么操作没有代码显示?答案是有,
def Count_SubDir(src): listdir = [] listall = os.listdir(src) for line in listall: filepath = os.path.join(src,line) if os.path.isdir(filepath): listdir.append(filepath) return listdir
输出结果为:
Type "copyright", "credits" or "license()" for more information. >>> ================================ RESTART ================================ >>> f:\src\1025 f:\src\1028 f:\src\1029 f:\src\1030 f:\src\1031 f:\src\1032 f:\src\1033 f:\src\1035 f:\src\1036 f:\src\1037 f:\src\1038 f:\src\1040 f:\src\1041 f:\src\1042 f:\src\1043 f:\src\1044 f:\src\1045 f:\src\1046 f:\src\1049 f:\src\1053 f:\src\1055 f:\src\2052 f:\src\2070 f:\src\3076 f:\src\3082 f:\src\Graphics >>>
好了,这次真的完整了,是真的目录了。。。。不敢信,一开始没找到问题的时候,总感觉没问题。。。不分析结果还真把remove方法的末尾处理给漏掉了。。。我想不少人也只是知道remove的删除这一步吧,没有想到删除之后的列表调整吧,哈哈!
时间: 2024-10-04 18:11:20