相关疑难解决方法(0)

是否可以将此代码转换为列表理解？

for i in userInput:
    if i in wordsTask:
        a = i
        break

我知道如何转换它的一部分:

[i for i in userInput if i in wordsTask]

但我不知道如何添加休息,文档也没有多大帮助.

任何帮助,将不胜感激.

我正在尝试遍历文件夹和所有子文件夹以查找某些文件类型的所有文件 - 例如,仅.mp4,.avi,.wmv.

这是我现在拥有的,它遍历所有文件类型:

import os
rootdir = 'input'

for subdir, dirs, files in os.walk(rootdir):
     for file in files:
          print (os.path.join(subdir, file))

谢谢!