Регулярное выражение, чтобы соответствовать число без определенного суффикса - PullRequest
0 голосов
/ 10 апреля 2019

У меня есть следующие три строки:

inputs = [
    "Season 12",
    "Season 1a",
    "Season 1-2"
]

Я пытаюсь соответствовать только первому. Вот текущее регулярное выражение, которое у меня есть:

outputs = []
for input in inputs:
    output = re.search(r'(Staffel|Season|Saison|S\.?)?\s?(\d{0,})(?!(-|[a-z][A-Z]))', input, re.IGNORECASE).group(2)
    outputs.append(output)

assert(outputs == ['12','',''])
# AssertionError, values were ['12', '1', '']

В настоящее время это работает для Season 12, Season 1-2, но не для Season 1a (ничего не должно возвращаться).

Ответы [ 2 ]

1 голос
/ 10 апреля 2019

Не уверен, что вы хотите за Seinfeld, Season 1 (UHD).Здесь это захватывает, но если вы не хотите, измените последнюю часть с (?:\s|$) на простой $

import re

inputs = [
    "Season 12",
    "Season 1a",
    "Season 1-3",
    "Seinfeld, Season 1 (UHD)",
    "Seinfeld, Season 1"
]

outputs = []
for input in inputs:
    output = re.search(r'(?:Staffel|Season|Saison|S\.?)?\s(\d+)(?:\s|$)', input, re.IGNORECASE)
    if (output != None):
        outputs.append(output.group(1))
    else:
        outputs.append('')

print(outputs)
assert(outputs == ['12','','','1','1'])

Вывод:

['12', '', '', '1', '1']
1 голос
/ 10 апреля 2019
import re
inputs = [
    "Season 12",
    "Season 1a",
    "Season 1-2",
    "Seinfeld, Season 1 (UHD)"
]
re_num = re.compile(
    r'(Staffel|Season|Saison|S\.?)\s?((\d+)$|(\d+)\s)',
    flags=re.IGNORECASE
)

for s in inputs:
    m = re_num.search(s)
    if m:
        print(s, '-->', m.group(2))

результат:

Season 12 --> 12
Seinfeld, Season 1 (UHD) --> 1
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...