拆分一个字符串并将子字符串保存到dict.Python

Question

我有这样一个文本文件：

771 776 #1 556.766700(2)
538 #2 1069.652700(2)
531 #3 1074.407600(2)
81 84 89 94 111 #4 1501.062900(2)
85 91 #5 782.298900(3)
32 42 66 71 90 95 101 #6 904.016500(3)

我想将子字符串拆分并保存到不同的变量中，如下所示： 例如第1行：

scans= 771 776, uid = 1 mz = 556.766700, z = 2

我正在尝试使用以下代码，但需要有关regex的帮助：

f = open(filename, 'r')
par_info=[]
for rows in f:
    re.sub('\#(.+)\s(.+)\((.+)\+', scans=\g<1>, uid=\g<2>, mz = int(\g<3>),    z=int(\g<4>), rest)
    info={'sc_num':scans, 'ident':uid, 'mass':mz, 'charge':z}
    par_info.append(info)

Answer 1

可以使用命名组：

>>> import pprint
>>> import re
>>> r = re.compile(r'(?P<scans>.*?)\s+#(?P<uid>\d+)\s+(?P<mz>\d+\.\d+)\((?P<z>\d+)\)')
>>> with open('abc1') as f:
        par_info = [r.search(line).groupdict() for line in f]
...     
>>> pprint.pprint(par_info)
[{'mz': '556.766700', 'scans': '771 776', 'uid': '1', 'z': '2'},
 {'mz': '1069.652700', 'scans': '538', 'uid': '2', 'z': '2'},
 {'mz': '1074.407600', 'scans': '531', 'uid': '3', 'z': '2'},
 {'mz': '1501.062900', 'scans': '81 84 89 94 111', 'uid': '4', 'z': '2'},
 {'mz': '782.298900', 'scans': '85 91', 'uid': '5', 'z': '3'},
 {'mz': '904.016500', 'scans': '32 42 66 71 90 95 101', 'uid': '6', 'z': '3'}]

Answer 2

这个正则表达式工作，然后您可以将找到的组压缩到一起，并将它们转换为dict：

In [1]: import re

In [2]: a = "771 776 #1 556.766700(2)"

In [3]: c = re.compile(r'([\d\s]+)\s#(\d)+\s([\d\.]+)\((\d+)\)')

In [4]: titles = ('sc_num', 'ident', 'mass', 'charge')

In [5]: dict(zip(titles, c.search(a).groups()))
Out[5]: {'charge': '2', 'ident': '1', 'mass': '556.766700', 'sc_num': '771 776'}

把这些和你的代码放在一起你会得到：

f = open(filename, 'r')
c = re.compile(r'([\d\s]+)\s#(\d)+\s([\d\.]+)\((\d+)\)')
titles = ('sc_num', 'ident', 'mass', 'charge')
par_info=[]
for row in f:
    info = dict(zip(titles, c.search(row).groups()))
    par_info.append(info)

Answer 3

import re
pattern = re.compile("(\d+\s*\d*)\s+#(\d+)\s+([\d\.]+)\s*\((\d+)\)")
for line in open("Input.txt"):
    scans, uid, mz, z = pattern.findall(line)[0]
    print scans, uid, mz, z

输出

771 776 1 556.766700 2
538 2 1069.652700 2
531 3 1074.407600 2
94 111 4 1501.062900 2
85 91 5 782.298900 3
95 101 6 904.016500 3

正则表达式演示

Debuggex Demo