Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Python Regex to Parse String and Return Tuple

I've been given some strings to work with. Each one represents a data set and consists of the data set's name and the associated statistics. They all have the following form:

s= "| 'TOMATOES_PICKED'                                  |       914 |       1397 |"

I'm trying to implement a function that will parse the string and return the name of the data set, the first number, and the second number. There are lots of these strings and each one has a different name and associated stats so I've figured the best way to do this is with regular expressions. Here's what I have so far:

def extract_data2(s):
    import re
    name=re.search('\'(.*?)\'',s).group(1)
    n1=re.search('\|(.*)\|',s)
    return(name,n1,)

So I've done a bit of reading on regular expressions and figured out how to return the name. For each of the strings that I'm working with, the name of the data set is bounded by ' ' so that's how I found the name. That part works fine. My problem is with getting the numbers. What I'm thinking right now is to try to match a pattern that is preceded by a vertical bar ('|'), then anything (which is why I used .*), and followed by another vertical bar to try to get the first number. Does anyone know how I can do this in Python? What I tried in the above code for the first number returns basically the whole string as my output, whereas I want to get just the number. -I am very new to programming so I apologize if this question seems rudimentary, but I have been reading and searching quite diligently for answers that are close to my case with no luck. I appreciate any help. The idea is that it will be able to:

return(name,n1,n2)

so that when the user inputs a string, it can just parse up the string and return the important information. I've noticed in my attempts to get the numbers so far that it will return the number as a string. Is there anyway to return n1 or n2 as just a number? Note that for some of the strings n1 and n2 could be either integers or have a decimal.

like image 846
Simos Anderson Avatar asked Jun 07 '11 05:06

Simos Anderson


People also ask

What will the '$' regular expression match?

By default, regular expressions will match any part of a string. It's often useful to anchor the regular expression so that it matches from the start or end of the string: ^ matches the start of string. $ matches the end of the string.

What is regex and parsing in Python?

A RegEx, or Regular Expression, is a sequence of characters that forms a search pattern. RegEx can be used to check if a string contains the specified search pattern.

What does re match return Python?

Both return the first match of a substring found in the string, but re. match() searches only from the beginning of the string and return match object if found. But if a match of substring is found somewhere in the middle of the string, it returns none.


1 Answers

I would use a single regular expression to match the entire line, with the parts I want in named groups ((?P<name>exampl*e)).

import re
def extract_data2(s):
    pattern = re.compile(r"""\|\s*                 # opening bar and whitespace
                             '(?P<name>.*?)'       # quoted name
                             \s*\|\s*(?P<n1>.*?)   # whitespace, next bar, n1
                             \s*\|\s*(?P<n2>.*?)   # whitespace, next bar, n2
                             \s*\|""", re.VERBOSE)
    match = pattern.match(s)

    name = match.group("name")
    n1 = float(match.group("n1"))
    n2 = float(match.group("n2"))

    return (name, n1, n2)

To convert n1 and n2 from strings to numbers, I use the float function. (If they were only integers, I would use the int function.)

I used the re.VERBOSE flag and raw multiline strings (r"""...""") to make the regex easier to read.

like image 112
Jeremy Avatar answered Sep 22 '22 04:09

Jeremy