How to extract a number before a certain words?

Question

There is a sentence "i have 5 kg apples and 6 kg pears".

I just want to extract the weight of apples.

So I use

sentence = "I have 5 kg apples and 6 kg pears"
number = re.findall(r'(\d+) kg apples', sentence)
print (number)

However, it just works for integer numbers. So what should I do if the number I want to extract is 5.5?

Answer 1

You can try something like this:

import re

sentence = ["I have 5.5 kg apples and 6 kg pears",
                   "I have 5 kg apples and 6 kg pears"]
for sen in sentence:
    print re.findall(r'(\d+(?:\.\d+)?) kg apples', sen)

Output:

['5.5']
['5']