python - 在字符串中搜索唯一的单词

Question

我几乎编写了一个可以在 GEDCOM 文件(保存有关家谱信息的文件)上执行多项任务的程序。

不过我只面临一个问题；有一些像这样的行:

0 @F8@ FAM
1 FAMC @F5@
1 FAMS @F5@

现在我想提取以@开头并以@结尾的代码。

我用过:

 if 'FAM' in line:
     var = line[1:6]
     ...

但是当我运行该程序时，它会输出带有 FAMC 和 FAMS 的所有行，因为这些单词内部有 FAM 。

如何提取仅匹配 FAM 的行？谢谢。

Answer 1

您可以使用正则表达式(字边界)来完成此操作，并且 re.search() :

lines = [
    "0 @F8@ FAM",
    "1 FAMC @F5@",
    "1 FAMS @F5@"
]

for line in lines:
    if re.search(r'\bFAM\b', line):
        var = line
        print var

“单词边界”(\b) 标记单词的开始或结束。

我们还可以同时使用re.search()来提取家族代码:

for line in lines:
    search = re.search(r'@([A-Z0-9]+)@\s*\bFAM\b', line)
    if search:
        code = search.group(1)
        print code