Question

import glob

list = glob.glob(r'*abc*.txt') + glob.glob(r'*123*.txt') + glob.glob(r'*a1b*.txt')

for i in list:
  print i

此代码用于列出当前文件夹中名称中包含“abc”，“123”或“a1b”的文件。

我如何使用一个glob来执行此功能？谢谢！

Answer 1

最简单的方法是自己过滤glob结果。以下是使用简单循环理解的方法：

import glob
res = [f for f in glob.glob("*.txt") if "abc" in f or "123" in f or "a1b" in f]
for f in res:
    print f

你也可以使用正则表达式而不是glob：

import os
import re
res = [f for f in os.listdir(path) if re.search(r'(abc|123|a1b).*\.txt$', f)]
for f in res:
    print f

（顺便说一句，命名变量list是一个坏主意，因为list是Python类型......）

Answer 2

根据其他答案，这是一种随时可用的方法。它不是最重要的性能，但它的工作原理如上所述;

def reglob(path, exp, invert=False):
    """glob.glob() style searching which uses regex

    :param exp: Regex expression for filename
    :param invert: Invert match to non matching files
    """

    m = re.compile(exp)

    if invert is False:
        res = [f for f in os.listdir(path) if m.search(f)]
    else:
        res = [f for f in os.listdir(path) if not m.search(f)]

    res = map(lambda x: "%s/%s" % ( path, x, ), res)
    return res

Answer 3

我很惊讶这里没有答案使用过滤器。

import os
import re

def glob_re(pattern, strings):
    return filter(re.compile(pattern).match, strings)

filenames = glob_re(r'.*(abc|123|a1b).*\.txt', os.listdir())

这接受任何返回字符串的迭代器，包括列表，元组，字典（如果所有键都是字符串）等。如果要支持部分匹配，可以将.match更改为.search。请注意，这显然会返回一个生成器，因此，如果您要使用结果而不对其进行迭代，则可以将结果自己转换为列表，或者将return语句包装为list（...）。

Answer 4

for filename in glob.iglob(path_to_directory + "*.txt"):
    if filename.find("abc") != -1 or filename.find("123") != -1 or filename.find("a1b") != -1:
        print filename

glob.glob中用于python的正则表达式用法

4 个答案: