Python正则表达式字符串扩展

Mark Kennedy 发表于 Dev

马克·肯尼迪

假设我有以下字符串：

trend  = '(A|B|C)_STRING'

我想将此扩展为：

A_STRING
B_STRING
C_STRING

OR条件可以在字符串中的任何位置。即STRING_(A|B)_STRING_(C|D)

将扩展到

STRING_A_STRING_C
STRING_B_STRING C
STRING_A_STRING_D
STRING_B_STRING_D

我还想介绍一个空条件的情况：

(|A_)STRING 将扩展为：

A_STRING
STRING

到目前为止，这是我尝试过的方法：

def expandOr(trend):
    parenBegin = trend.index('(') + 1
    parenEnd = trend.index(')')
    orExpression = trend[parenBegin:parenEnd]
    originalTrend = trend[0:parenBegin - 1]
    expandedOrList = []

    for oe in orExpression.split("|"):
        expandedOrList.append(originalTrend + oe)

但这显然行不通。

使用正则表达式有什么简单的方法吗？

蒂姆·彼得斯

这是一种非常干净的方法。弄清楚它是如何工作的，您将很开心:-)

def expander(s):
    import re
    from itertools import product
    pat = r"\(([^)]*)\)"
    pieces = re.split(pat, s)
    pieces = [piece.split("|") for piece in pieces]
    for p in product(*pieces):
        yield "".join(p)

然后：

for s in ('(A|B|C)_STRING',
          '(|A_)STRING',
          'STRING_(A|B)_STRING_(C|D)'):
    print s, "->"
    for t in expander(s):
        print "   ", t

显示：

(A|B|C)_STRING ->
    A_STRING
    B_STRING
    C_STRING
(|A_)STRING ->
    STRING
    A_STRING
STRING_(A|B)_STRING_(C|D) ->
    STRING_A_STRING_C
    STRING_A_STRING_D
    STRING_B_STRING_C
    STRING_B_STRING_D

本文收集自互联网，转载请注明来源。

如有侵权，请联系[email protected] 删除。