时间戳

首页 > 解决方案 > 使用 re.sub 替换字符串的数字部分，并在 Python 中对该数字进行算术操作？

python - 使用 re.sub 替换字符串的数字部分，并在 Python 中对该数字进行算术操作？

问题描述

假设我有字符串testing-1-6-180- 这里我想捕获第二个数字（不管它是什么），这里是“6”，然后我想将 5 添加到它的数值（所以 6），然后输出字符串 - 所以在这种情况下，结果应该是testing-1-11-180.

这是我到目前为止所尝试的：

import re

mytext = "testing-1-6-180"
pat_a = re.compile(r'testing-1-(\d+)')
result = pat_a.sub( "testing-1-{}".format( int('\1')+5 ), mytext )

...不幸的是，这失败了：

$ python3 test.py
Traceback (most recent call last):
  File "test.py", line 7, in <module>
    result = pat_a.sub( "testing-1-{}".format( int('\1')+5 ), mytext )
ValueError: invalid literal for int() with base 10: '\x01'

那么，如何获取捕获的反向引用，以便将其转换为 int，进行一些算术运算，然后使用结果替换匹配的子字符串？

能够发布答案会很好，因为在这里弄清楚如何将答案应用于此问题并不是一件容易的事，但无论如何没人关心，所以我将答案发布为编辑：

import re

mytext = "testing-1-6-180"
pat_a = re.compile(r'testing-1-(\d+)')

def numrepl(matchobj):
  return "testing-1-{}".format( int(matchobj.group(1))+5 )

result = pat_a.sub( numrepl, mytext )
print(result)

结果是testing-1-11-180。

标签： pythonregexarithmetic-expressionsbackreference

解决方案

您可以使用 lambda 代替：

>>> mytext = "testing-1-6-180"
>>> s = re.sub(r'^(\D*\d+\D+)(\d+)', lambda m: m.group(1) + str(int(m.group(2)) + 5), mytext)
>>> print (s)
'testing-1-11-180'

推荐阅读