筛选任何包含从其他列表

Question

我想缩短这个函数的任何项目的任何部分列表：

def get_filenames(path, banned_files=()): 
    file_list = [] 
    for root, _, files in os.walk(path): 
     BANNED_FILES = banned_files 
     for f in files: 
      for string in BANNED_FILES: 
       if string in f: 
        continue 
      path = os.path.join(root, f) 
      file_list.append(path) 
    return file_list 

这样使用：

filenames = get_filenames(CLIENT_TESTS_PATH, banned_files=['__init__.py', '.pyc', 'accounts.py, otherfile.py']) 

使得从任何文件CLIENT_TESTS_PATH没有任何被禁止的文件。我不能仅仅检查文件路径是否在带有列表理解的禁止文件中，因为我关心是否可以在客户端文件中的任何位置找到禁止的文件。我怎样才能缩短这与一个过滤器，列表理解和或lambda有效？我只想要那个目录下的文件，而不是下面的目录。 谢谢

对于

def get_files(path, banned_files=[]): 
    return [os.path.join(root, fname) for root, _, fnames in os.path.walk(path) for fname in fnames \ 
       if not any(bad_fname in fname for bad_fname in banned_files)] 

我得到

In [2]: CLIENT_TESTS_PATH = "/home/cchilders/work_projects/webapi"  

In [3]: get_files(CLIENT_TESTS_PATH) 
--------------------------------------------------------------------------- 
TypeError         Traceback (most recent call last) 
<ipython-input-3-3c5a75968d68> in <module>() 
----> 1 get_files(CLIENT_TESTS_PATH) 

/home/cchilders/scripts/my_scripting_library/sysadmin.py in get_files(path, banned_files) 
    15 
    16 def get_files(path, banned_files=[]): 
---> 17  return [os.path.join(root, fname) for root, _, fnames in os.path.walk(path) for fname in fnames \ 
    18     if not any(bad_fname in fname for bad_fname in banned_files)] 
    19 

TypeError: walk() takes exactly 3 arguments (1 given) 

Answer 1

喜欢的东西：

banned_files=['__init__.py', '.pyc', 'accounts.py', 'otherfile.py'] 
file_list = ["""Some list of files here"""] 
new_list = {file_name for file_name in file_list for bf in banned_files if bf not in file_name} 

这将过滤掉任何东西，都在文件名中这些字符串。这意味着像"my_accounts.py.txt"将被阻止。将它们分开可能会更好。例如：

banned_files=['__init__.py', 'accounts.py', 'otherfile.py'] 
banned_extensions = ['.pyc'] 
new_list = {file_name for file_name in file_list for ext in banned_extensions if file_name not in banned_files and not file_name.endswith(ext)} 

编辑包含os.walk ...一秒钟。

Answer 2

列表理解：

def get_filenames(path, banned_files=()): 
    return [os.path.join(root, f) for root,_,files in os.walk(path) for f in files] 

，做同样的事情，你发布的代码，但它可能是你想要的东西没有做。你的for string in BANNED_FILES:循环完全没用，因为它所做的只是继续，但continue只能在当前循环中使用;它不会影响for f in files:循环，因此除了浪费处理时间之外，内部循环完全不起作用。为了做我认为你想，这样做：

def get_filenames(path, banned_files=()): 
    return [os.path.join(root, f) 
      for root,_,files in os.walk(path) 
       for f in files 
        if not any(string in f for string in banned_files) 
    ] 

Answer 3

列表理解是恕我直言，最清晰的方式：

[os.path.join(root, f) for root, _, files in os.walk(path) \ 
for f in files if all([bf not in f for bf in banned_files])] 

不要忘记在所有的（）括号：

In [7]: [f for f in ['abc','def','ghi','jkl'] if all([bf not in f \ 
for bf in ['a','e','z']])] 
Out[7]: ['ghi', 'jkl'] 

In [8]: [f for f in ['abc','def','ghi','jkl'] if all(bf not in f \ 
for bf in ['a','e','z'])] 
Out[8]: ['abc', 'def', 'ghi', 'jkl'] 

Answer 4

我会做这种方式：

def get_filenames(path, banned_files=[]): 
    banned = '|'.join(banned_files) 
    return [os.path.join(root, f) 
      for root, _, files in os.walk(path) 
      for f in files 
      if f not in banned] 

说明：

|不能在文件名中使用，所以我们可以使用它作为一个包含所有被禁止的文件字符串分隔符。检查发现的文件名对单个字符串要快得多。

Answer 5

[os.path.join(root, fname) for root, _, fnames in os.path.walk(path) for fname in fnames \ 
    if not any(bad_fname in fname for bad_fname in banned)] 

Answer 6

试试这个：

def get_filenames(path, banned_files=()): 
    file_list = [os.path.join(root, f) for root, _, files in os.walk(path) 
       for f in files if all(s not in f for s in banned_files)] 
    return file_list