在Python列表中查找元素的有效方法？

Question

我有一个目录中的文件列表。我必须只处理该目录中的某些文件。 filelist 是我想要的文件列表。我该如何实现这一目标？对 bash 解决方案不感兴趣，因为我必须在这个 Python 脚本中完成这一切。非常感谢！

for record in result:
    filelist.append(record[0])

print filelist


for file in os.listdir(sys.argv[1].strip() + "/"):
    for file in filelist: #This doesn't work, how else do I do this? If file equals to my desired file-list, then do something.
        print file

抱歉各位，不知道我是怎么错过这个的！我猜是一大早编码！！模组们，请关闭它，除非有人想要以有效的方式进行操作。

for file in os.listdir(sys.argv[1].strip() + "/"):
    if file in filelist:
        print file

Answer 1

如果顺序和唯一性不重要，您可以使用 set 交集，这会更高效。

import set
os_list = os.listdir(sys.argv[1].strip() + "/")
for file in set(os_list) & set(filelist):
    #...

改进示例：

import random
import timeit

l = [random.randint(1,10000) for i in range(1000)]
l2 = [random.randint(1,10000) for i in range(1000)]

def f1():
    l3 = []
    for i in l:
        if i in l2:
            l3.append(i)
    return l3

def f2():
    l3 = []
    for i in set(l) & set(l2):
        l3.append(i)
    return l3

t1 = timeit.Timer('f1()', 'from __main__ import f1')
print t1.timeit(100) #2.0850549985

t2 = timeit.Timer('f2()', 'from __main__ import f2')
print t2.timeit(100) #0.0162533142857

Answer 2

听起来你想做一个测试：

for file in os.listdir(sys.argv[1].strip() + "/"):
    if file in filelist:
        # Found a file in the wanted-list.
        print file

Answer 3

看起来您只想做这样的事情：

for file in os.listdir(sys.argv[1].strip() + "/"): 
    if file in filelist:
        print file 

请注意，我刚刚将第二个 for 更改为 if。但是，由于您询问效率，您可能希望将 filelist 从 list 更改为 set 或 dict 使 in 运算符更加高效。

Answer 4

像这样的东西：

print [x for x in os.listdir(sys.argv[1].strip() + "/") if x in filelist]