I have a long Python list like:
['A p=45 n=200 SNR=12', 'B p=2232 n=22 SNR=2']
I would like to get from this list a list of tuples containing the value of p, n and SNR.
So:
funz(['A p=45 n=200 SNR=12', 'B p=2232 n=22 SNR=2'])
would return:
[(45,200,12), (2232,22,2)]
The strings in the list have all the same structure.
import re
data = ['A p=45 n=200 SNR=12', 'B p=2232 n=22 SNR=2']
result = []
for x in data:
result.append( map( int, re.search('p=(\d+).*n=(\d+).*SNR=(\d+)', x).groups()) )
print result
[[45, 200, 12], [2232, 22, 2]]
Another option is this:
def funz(l):
return [tuple(int(i.split('=')[1]) for i in item.split(' ')[1:]) for item in l]
Edited re @bvukelic's comment.
You can use re.findall and a list comprehension:
>>> from re import findall
>>> lst = ['A p=45 n=200 SNR=12', 'B p=2232 n=22 SNR=2']
>>> def funz(lst):
... return [tuple(map(int, findall("\d+", i))) for i in lst]
...
>>> funz(['A p=45 n=200 SNR=12', 'B p=2232 n=22 SNR=2'])
[(45, 200, 12), (2232, 22, 2)]
>>>
The Regex pattern \d+ matches one or more digits.
[tuple(int(m) for m in re.findall(r'(\d+)', i)) for i in lst]