Python如何查找字符串的结尾4个字符或5个字符是否遵循模式

Question

我有一堆需要清理的字符串，它有以下模式：

12345SNET1
1234567SNETA2
123456SNET3

令人头疼的是，SNET之后的任何内容都可以是从0到9的任何整数，也可以是A-Z的char加上从0到9的整数。

是否无论如何都要使用正则表达式来检测字符串是否具有此模式，以便我可以使用：

if regex detect (returns True):
    str = str[:-1]

elif regex detect (returns True):
    str = str[:-2]

Answer 1

如果使用正确的正则表达式，不需要有两个case。

values = ["12345SNET1", "1234567SNETA2", "123456SNET3"]
for value in values:
    m = re.match(r'\d+SNET([A-Z]?\d)', value)
    if m:
        print(m.group(1))

这会打印出来的

1
A2
3

如果希望文本位于最后一个字符之前，可以在正则表达式中添加额外的括号以捕捉该部分：

values = ["12345SNET1", "1234567SNETA2", "123456SNET3"]
for value in values:
    m = re.match(r'(\d+SNET)([A-Z]?\d)', value)
    if m:
        print(m.group(1))

后果

12345SNET
1234567SNET
123456SNET

Answer 2

您可以使用re.sub结合正后置来丢弃不需要的字符，方法如下：

import re
s1 = "12345SNET1"
s2 = "1234567SNETA2"
s3 = "123456SNET3"
out1 = re.sub(r"(?<=SNET)[A-Z]?\d", "", s1)
out2 = re.sub(r"(?<=SNET)[A-Z]?\d", "", s2)
out3 = re.sub(r"(?<=SNET)[A-Z]?\d", "", s3)
print(out1)  # 12345SNET
print(out2)  # 1234567SNET
print(out3)  # 123456SNET

Answer 3

>

if re.fullmatch(r".*SNET\d", value):
    value = value[:-1]

if re.fullmatch(r".*SNET[A-Z]\d", value):
    value = value[:-2]

您可以直接使用re.sub替换末尾

value = re.sub(r"(?<=SNET)[A-Z]?\d", "", value)

为了使用，您可以将其导出到方法中

def clean(value):
    if re.fullmatch(r".*SNET\d", value):
        return value[:-1]
    if re.fullmatch(r".*SNET[A-Z]\d", value):
        return value[:-2]
    return value

# OR
def clean(value):
    return re.sub(r"(?<=SNET)[A-Z]?\d", "", value)


if __name__ == '__main__':
    values = ["12345SNET1", "1234567SNETA2", "123456SNET3"]
    print(values)  # ['12345SNET1', '1234567SNETA2', '123456SNET3']
    values = list(map(clean, values))
    print(values)  # ['12345SNET', '1234567SNET', '123456SNET']