用python从不规则值中提取数字

Question

我有如下数据：

Format,Message,time
A,gn@2 ab@1 yl@5 rd@20 pp@40,3
B,w:w23w4w5w6w7gn@3 gn@7 yl@20 ss@25 rd@50,21
C,cc@1 fgn@4 yl@9 rd@20,22
D,rg@1 fedsf@5 rww@10 yl@20 rd@26,30

我的预期结果是提取gn、yl和rd之后的数字

Format,Message,time,gn,yl,rd
A,gn@2 ab@1 yl@5 rd@20 pp@40,3,2,5,20
B,w:w23w4w5w6w7gn@3 an@7 yl@20 ss@25 rd@50,21,3,20,50
C,cc@1 fgn@4 yl@9 rd@20,22,4,9,20
D,rg@1 fedsf@5 rww@10 yl@20 rd@26,30,0,20,26

到目前为止，我可以得到yl和rd，但我无法提取数字后面的gn。请注意，gn元素可能在gn之前包含一些其他字符，并且在gn@

def f(mess):
    p1 = mess.find('yl')
    p2 = mess.find('rd')
    b = mess[p1+3:].split(' ')[0]
    c = mess[p2+3:].split(' ')[0]
    return int(b),int(c)
id['vals'] = id['Message'].apply(f) #with this im able to get the numbers from yl and rd

Answer 1

使用正则表达式。你知道吗

演示：

import re
s = """A,gn@2 ab@1 yl@5 rd@20 pp@40,3,2,5,20
B,w:w23w4w5w6w7gn@3 an@7 yl@20 ss@25 rd@50,21,3,20,50
C,cc@1 fgn@4 yl@9 rd@20,22,4,9,20
C,cc@1 yl@9 rd@20,22,4,9,20"""

for line in s.splitlines():
    gn = re.search(r"gn@(.?\S)", line)
    if gn:
        gn = gn.group(1)

    yl = re.search(r"yl@(.?\S)", line)
    if yl:
        yl = yl.group(1)

    rd = re.search(r"rd@(.?\S)", line)
    if rd:
        rd = rd.group(1)
    print(gn, yl, rd)

输出：

2 5 20
3 20 50
4 9 20
None 9 20

Answer 2

尝试使用以下表达式：

mess = 'gn@2 ab@1 yl@5 rd@20 pp@40'
result = [ int(m.split('@')[1])  for m in mess.split() if m.split('@')[0] in ['gn', 'yl', 'rd'] ]

Answer 3

让我们分步来解决这个问题。你知道吗

1. 只获取您感兴趣的行。你知道吗
2. 删除可能对我们没有用处的数据。你知道吗
3. 使用剩下的数据提取信息。你知道吗

假设输入存储在变量data中，输出存储在名为final的元组列表中。下面是我将如何解决这个问题。你知道吗

useful = data.split('\n')[1:]  ## Step 1
code = [x[1].strip() for x in useful.split(',')] ## Step 2
gn_value = -1
yl_value = -1
rd_value = -1
for line in code:
    for each in line.split(' '): ## Step 3
        if 'gn@' in each:
            gn_value = int(each[each.find('gn@')+3:])
        elif 'yl@' in each:
            yl_value = int(each[each.find('yl@')+3:])
        elif 'rd@' in each:
            rd_value = int(each[each.find('rd@')+3:])
    final.append(gn_value, yl_value, rd_value)

注：上述解决方案是在假设任何给定行中没有任何值多次出现的情况下得出的。你知道吗

如果您有任何疑问，请告诉我。你知道吗