您可以将黑名单加入到一个表达式中:
import re
blacklist = re.compile('|'.join([re.escape(word) for word in B]))
然后过滤掉匹配的单词:
C = [word for word in A if not blacklist.search(word)]
模式中的单词被转义(因此. 和其他元字符不被视为这样,而是被视为文字字符),并加入一系列| 替代项:
>>> '|'.join([re.escape(word) for word in B])
'XXX|BBB'
演示:
>>> import re
>>> A = [ 'cat', 'doXXXg', 'monkey', 'hoBBBrse', 'fish', 'snake']
>>> B = ['XXX', 'BBB']
>>> blacklist = re.compile('|'.join([re.escape(word) for word in B]))
>>> [word for word in A if not blacklist.search(word)]
['cat', 'monkey', 'fish', 'snake']
这应该优于任何显式成员资格测试,尤其是随着黑名单中单词数量的增加:
>>> import string, random, timeit
>>> def regex_filter(words, blacklist):
... [word for word in A if not blacklist.search(word)]
...
>>> def any_filter(words, blacklist):
... [word for word in A if not any(bad in word for bad in B)]
...
>>> words = [''.join([random.choice(string.letters) for _ in range(random.randint(3, 20))])
... for _ in range(1000)]
>>> blacklist = [''.join([random.choice(string.letters) for _ in range(random.randint(2, 5))])
... for _ in range(10)]
>>> timeit.timeit('any_filter(words, blacklist)', 'from __main__ import any_filter, words, blacklist', number=100000)
0.36232495307922363
>>> timeit.timeit('regex_filter(words, blacklist)', "from __main__ import re, regex_filter, words, blacklist; blacklist = re.compile('|'.join([re.escape(word) for word in blacklist]))", number=100000)
0.2499098777770996
上面测试了 10 个随机列入黑名单的短词(2 - 5 个字符)与 1000 个随机词(3 - 20 个字符长)的列表,正则表达式的速度大约快 50%。