python - 如何在 python 中获取一个单词后的数字

Question

我有一个巨大的文件，其中包含以下行 DDD-1126N|refseq:NP_285726|uniprotkb:P00112 和 DDD-1081N|uniprotkb:P12121，我想捕获uniprotkb 之后的数字。

这是我的代码:

x = 'uniprotkb:P'
f = open('m.txt')
for line in f:
  print line.find(x) 
  print line[36:31 + len(x)]

line.find(x)中的问题是10和26，我是26的时候抓取完整的数字，我是编程新手，所以找东西抓取单词后的完整数字。

x = 'uniprotkb:'
f = open('m.txt')
for line in f:
  if x in line:
    print the number after x

Answer 1

使用正则表达式:

import re
for line in open('m.txt'):
    match = re.search('uniprotkb:P(\d+)', line)
    if match:
        print match.group(1)