时间戳

首页 > 解决方案 > 有没有办法在不同的替代分支中统一正则表达式捕获组？

python - 有没有办法在不同的替代分支中统一正则表达式捕获组？

问题描述

假设我有这样的代码：

import re

rx = re.compile(r'(?:(\w{2}) (\d{2})|(\w{4}) (\d{4}))')
def get_id(s):
    g = rx.match(s).groups()
    return (
        g[0] if g[0] is not None else g[2],
        int(g[1] if g[1] is not None else g[3]),
    )

print(get_id('AA 12'))      # ('AA', 12)
print(get_id('BBBB 1234'))  # ('BBBB', 1234)

这可以满足我的要求，但它需要我检查每个捕获组以检查哪个捕获组实际捕获了子字符串。如果替代品的数量很多，这可能会变得笨拙，所以我宁愿避免这种情况。

我尝试使用命名捕获，但(?:(P<s>\w{2}) (?P<id>\d{2})|(?P<s>\w{4}) (?P<id>\d{4}))只会引发错误。

在正则表达式中针对多个案例统一捕获组的答案中的技巧不起作用，因为(\w{2}(?= \d{2})|\w{4}(?= \d{4})) (\d{2}|\d{4})将捕获错误数量的数字，并且由于我不想进入的原因，我无法手动优化替代方案的顺序。

有没有更惯用的方式来写这个？

标签： pythonidiomsrecapture-group

解决方案

好像有！从re文档：

        (?(id/name)yes|no) Matches yes pattern if the group with id/name matched,
                           the (optional) no pattern otherwise.

这使得这个例子：

rx = re.compile(r'(?P<prefix>(?P<prefix_two>)\w{2}(?= \d{2})|(?P<prefix_four>)\w{4}(?= \d{4})) (?P<digits>(?(prefix_two))\d{2}|(?(prefix_four))\d{4})')
def get_id(s):
    m = rx.match(s)
    if not m:
        return (None, None,)
    return m.group('prefix', 'digits')

print(get_id('AA 12'))      # ('AA', 12)
print(get_id('BB 1234'))    # ('BB', 12)
print(get_id('BBBB 12'))    # (None, None)
print(get_id('BBBB 1234'))  # ('BBBB', 1234)

是否值得麻烦，我会留给读者。

推荐阅读