我有这个代码:
x.write("FolderName\t\"%s\"\nPackName\t\"%s\"\n"% (pakedfolder, filename))
x.write("\nList\t%s\n{\n" % (comptype))
for root, dirs, files in os.walk("ymir work"):
for file in files:
file = path.splitext(file)[1]
x.write("\t\"%s\"\n" %(file))
x.write("}\n")
x.write("\nList\tFilelist\n{\n")
它生成的正是我想要的,但问题是代码中有重复,如下所示:
".dds"
".mse"
".mse"
".mse"
".mse"
".mse"
".mse"
".dds"
".dds"
".dds"
最佳答案
有更多可能的解决方案和途径来解决这个问题。
大多数人(以及 SO)都同意使用字典是正确的方法。
例如,史蒂夫布。 :D
有些人会认为 set() 会更方便、更自然,但我看到的和我自己做的大多数测试表明,出于某种原因,使用 dict() 稍微快一些。至于为什么,没有人真正知道。此外,这可能因 Python 版本而异。
字典和集合使用哈希来访问数据,这使得它们比列表更快( O(1) )。要检查某个项目是否在列表中,需要对列表执行迭代,在最坏的情况下,迭代次数会随着列表的增加而增长。
要了解有关该主题的更多信息,我建议您检查相关问题,尤其是提到的可能重复的问题。
因此,我同意 steveb 的观点并提出以下代码:
chkdict = {} # A dictionary that we'll use to check for existance of an entry (whether is extension already processed or not)
setdef = chkdict.setdefault # Extracting a pointer of a method out of an instance may lead to faster access, thus improving performance a little
# Recurse through a directory:
for root, dirs, files in os.walk("ymir work"):
# Loop through all files in currently examined directory:
for file in files:
ext = path.splitext(file) # Get an extension of a file
# If file has no extension or file is named ".bashrc" or ".ds_store" for instance, then ignore it, otherwise write it to x:
if ext[0] and ext[1]: ext = ext[1].lower()
else: continue
if not ext in chkdict:
# D.setdefault(k[, d]) does: D.get(k, d), also set D[k] = d if k not in D
# You decide whether to use my method with dict.setdefault(k, k)
# Or you can write ext separately and then do: chkdict[ext] = None
# Second solution might even be faster as setdefault() will check for existance again
# But to be certain you should run the timeit test
x.write("\t\"%s\"\n" % setdef(ext, ext))
#x.write("\t\"%s\"\n" % ext)
#chkdict[ext] = None
del chkdict # If you're not inside a function, better to free the memory as soon as you can (if you don't need the data stored there any longer)
我在大量数据上使用了这个算法,并且效果非常好。
关于python - 如何提取扩展名并保存而不重复?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/31895321/