Как увеличить число внутри строки с сохранением ведущих нулей?

Question

[Миша Тарасов]

Надо написать функцию, которая увеличивает строку, чтобы создать новую строку.

Если строка уже оканчивается цифрой, номер следует увеличить на 1.
Если строка не оканчивается цифрой. цифру 1 нужно добавить к новой строке.
Примеры:

foo -> foo1

foobar23 -> foobar24

foo0042 -> foo0043

foo9 -> foo10

foo099 -> foo100

Если число имеет ведущие нули, следует учитывать количество цифр.

Comments

[Hcuy]

Что именно не понятно?

[Миша Тарасов]

Hcuy, просто не понятно как решить эту задачу

Answer 1

[0xD34F]

def replacer(m):
  val = m.group(0)
  return str(int(val or '0') + 1).rjust(len(val), '0')

def increment(s):
  return re.sub(r'\d*$', replacer, s, 1)

Comments

[Алан Гибизов]

Ё-маё. Я голову сломал. Бедный препод, интересно, он вообще поймёт, что ему ученик принес на проверку?

Скажите, как Вы научились ТАК мыслить? Потрясен.

Я так что-то и не понял, что это за объект -_sre (я так понял, что это из c-шного содержания модуля re, но что это именно...) и как он попадает внутрь функции replacer.

Вывод:
Учиться, учиться и учиться!

p.s. Я ради интересу накарябал свой вариант решения задачи, но блин... он как паровоз Черепанова рядом с космической ракетой...
Правда, регулярку не использовал.

[0xD34F]

Алан Гибизов,

не понял, что это за объект -_sre

Я, признаться, тоже. Не вижу у себя никаких _sre.

[Алан Гибизов]

0xD34F, ну я стал ковырять этот самый m, смотреть, что туда попадает. А там унутре _sre.SRE_Match object

[0xD34F]

Алан Гибизов, это match object.

[Алан Гибизов]

0xD34F, но главное - как функция подставляется вместо параметра и откуда берет собственный параметр? Это где можно почитать?

[0xD34F]

Алан Гибизов, "как функция подставляется" - так же, как и всё остальное. Параметр передаёт sub - в качестве замены можно указывать функцию, которая получает match object и возвращает строку.

[Алан Гибизов]

0xD34F, вот sub ищет паттерн «строка цифр» во входной строке; находит. Заменяет на функцию. Но где и как происходит передача этой функции параметра m, который match-объект? Я чувствую, что оно где-то неявно передаётся, вроде как при применении map. Но как мне понять при написании своего кода, что оно должно туда передаваться?

[0xD34F]

Алан Гибизов,

Заменяет на функцию.

Нет. Заменяет на то, что функция вернёт.

где и как происходит передача этой функции параметра m

Где-то внутри sub.

Как - повторяю, так же, как и всё остальное. Как числа, строки и т.д. Или вас детали интересуют? - понятия не имею, да это и неважно.

как мне понять при написании своего кода, что оно должно туда передаваться?

Это ОЧЕНЬ странный вопрос. На который мне ответить нечего. Вот вы выше упомянули map, применительно к оному ваш вопрос звучал бы так: как мне понять, что в переданную ему (map'у) функцию должны передаваться элементы обрабатываемого списка/строки/...?

[Viktor T2]

Алан Гибизов,
Неявно вызывается.
re.sub(pattern, repl, string, count=0, flags=0)
Возвращает строку, полученную путем замены совпадения шаблона pattern в строке на замену repl.
repl может быть строкой или функцией;

Если repl является функцией, она вызывается для каждого совпадения шаблона.
Функция принимает Match Object аргумент, и возвращает строку замены.

Из оффициальной документации на re.
Но это конешно пипец эзотерика.