<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-12-08 05:34:32