python - 从python字符串中以简单的方式提取数字

Question

例如，我有一个存储数字和单位的字符串

x= '$120'
y = ' 90 Degrees F'
banana = '200 kgm'
orange = '300 gm'
total_weight = banana + orange/1000 

例如我想添加权重

total_weight  = 200 + 300/1000

谢谢！

我试图提取数字只是为了对这些进行一些操作......知道最简单的方法是什么吗？我只处理这两种格式，即数字位于字符串的开头或结尾...

Answer 1

从字符串中提取数字的最简单方法是使用正则表达式和findall.

>>> import re
>>> s = '300 gm'
>>> re.findall('\d+', s)
['300']
>>> s = '300 gm 200 kgm some more stuff a number: 439843'
>>> re.findall('\d+', s)
['300', '200', '439843']

您可能需要更复杂的东西，但这是一个很好的第一步。

请注意，您仍然需要调用int结果才能获得正确的数字类型（而不是另一个字符串）：

>>> map(int, re.findall('\d+', s))
[300, 200, 439843]

Answer 2

不使用regex，你可以这样做：

def get_num(x):
    return int(''.join(ele for ele in x if ele.isdigit()))

结果：

>>> get_num(x)
120
>>> get_num(y)
90
>>> get_num(banana)
200
>>> get_num(orange)
300

编辑 ：

回答后续问题。

如果我们知道给定字符串中唯一的句点是小数点，那么提取浮点数就很容易了：

def get_num(x):
    return float(''.join(ele for ele in x if ele.isdigit() or ele == '.'))

结果：

>>> get_num('dfgd 45.678fjfjf')
45.678

Answer 3

这个正则表达式也处理浮点数

import re
re_float = re.compile(r'\d*\.?\d+')

您还可以将一个组添加到捕获您的体重单位的表达式中。

re_banana = re.compile(r'(?P<number>\d*\.?\d+)\s?(?P<uni>[a-zA-Z]+)')

您可以像这样访问命名组re_banana.match("200 kgm").group('number')。

我认为这应该可以帮助您入门。

Answer 4

>>> x='$120'
>>> import string
>>> a=string.maketrans('','')
>>> ch=a.translate(a, string.digits)
>>> int(x.translate(a, ch))
120

Answer 5

如果您正在对数字进行某种数学运算，您可能还想知道单位。鉴于您的输入限制（输入字符串仅包含单位和值），这应该正确返回两者（您只需要弄清楚如何将单位转换为数学的常用单位）。

def unit_value(str):
    m = re.match(r'([^\d]*)(\d*\.?\d+)([^\d]*)', str)
    if m:
        g = m.groups()
        return ' '.join((g[0], g[2])).strip(), float(g[1])
    else:
        return int(str)