使用python的glob.glob中的正则表达式

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（...）。