Python利用re模块实现简易分词(tokenization)

慢慢来
慢慢来
发布时间:2026-10-15| 6人浏览过
原创

本文介绍如何使用Python的re模块构建简易分词器。通过定义命名捕获组匹配标识符、运算符和数字,利用scanner和match方法解析字符串。重点讲解模式顺序对子串匹配的影响及过滤空格的技巧,并推荐PyParsing等专业库处理复杂语法。

一个简单的tokenizer

分词(tokenization)是Python字符串处理中极为常见的任务。本文将演示如何利用正则表达式构建一个简易的表达式分词器(tokenizer),该工具能够将表达式字符串从左至右解析为标记(tokens)序列。

假设给定如下表达式字符串:

text = 'foo = 12 + 5 * 6'

我们的目标是将上述字符串转换为以下以序列对形式呈现的分词结果:

tokens = [('NAME', 'foo'), ('EQ', '='), ('NUM', '12'), ('PLUS', '+'),\
    ('NUM', '5'), ('TIMES', '*'), ('NUM', '6')]

要实现此类分词操作,首先需定义所有可能的标记模式(pattern)。所谓模式,即用于描述或匹配特定句法规则的字符串,此处我们使用正则表达式作为模式。需注意,模式中必须包含空格(whitespace)匹配,否则当字符串中出现未定义模式的字符时,扫描过程将会停止。此外,为了赋予标记如NAME、EQ等语义名称,我们采用正则表达式中的命名捕获组来实现。

import re
NAME = r'(?P<NAME>[a-zA-Z_][a-zA-Z_0-9]*)' 
# 这里?P<NAME>表示模式名称,()表示一个正则表达式捕获组,合在一起即一个命名捕获组
EQ = r'(?P<EQ>=)'
NUM = r'(?P<NUM>\d+)' #\d表示匹配数字,+表示任意数量
PLUS = r'(?P<PLUS>\+)' #需要用\转义
TIMES = r'(?P<TIMES>\*)' #需要用\转义
WS = r'(?P<WS>\s+)' #\s表示匹配空格, +表示任意数量
master_pat = re.compile("|".join([NAME, EQ, NUM, PLUS, TIMES, WS]))  # | 用于选择多个模式,表示"或"

随后,利用模式对象中的scanner()方法执行分词操作,该方法用于创建一个扫描对象:

scanner = master_pat.scanner(text)

接着,可以使用match()方法获取单次匹配结果,每次匹配一个模式:

scanner = master_pat.scanner(text)
m = scanner.match() 
print(m.lastgroup, m.group()) # NAME foo
m = scanner.match()
print(m.lastgroup, m.group()) # WS

显然,反复调用上述方法较为繁琐。我们可以利用迭代器进行批量调用,并将每次迭代的结果存储为具名元组形式。

Token = namedtuple('Token', ['type', 'value'])
def generate_tokens(pat, text):
    scanner = pat.scanner(text)
    for m in iter(scanner.match, None):
        #scanner.match做为迭代器每次调用的方法,
        #None为哨兵的默认值,表示迭代到None停止
        yield Token(m.lastgroup, m.group())
    
for tok in generate_tokens(master_pat, "foo = 42"):
    print(tok)

最终,表达式串"foo = 12 + 5 * 6"的tokens流显示如下:

Token(type='NAME', value='foo')
Token(type='WS', value=' ')
Token(type='EQ', value='=')
Token(type='WS', value=' ')
Token(type='NUM', value='12')
Token(type='WS', value=' ')
Token(type='PLUS', value='+')
Token(type='WS', value=' ')
Token(type='NUM', value='5')
Token(type='WS', value=' ')
Token(type='TIMES', value='*')
Token(type='WS', value=' ')
Token(type='NUM', value='6')

过滤tokens流

接下来,我们希望过滤掉空格标记,只需使用生成器表达式即可:

tokens = (tok for tok in generate_tokens(master_pat, "foo = 12 + 5 * 6")
          if tok.type != 'WS')
for tok in tokens:
    print(tok)

结果显示,空格已成功被过滤:

Token(type='NAME', value='foo')
Token(type='EQ', value='=')
Token(type='NUM', value='12')
Token(type='PLUS', value='+')
Token(type='NUM', value='5')
Token(type='TIMES', value='*')
Token(type='NUM', value='6')

注意子串匹配陷阱

在正则表达式(即"|".join([NAME, EQ, NUM, PLUS, TIMES, WS]))中,tokens的顺序至关重要。因为在进行匹配时,re模块会按照指定的顺序对模式进行匹配。因此,如果某个模式恰好是另一个较长模式的子串,必须确保较长模式排在前面以优先匹配。以下分别展示了正确的和错误的匹配方法:

LT = r'(?P<LT><)'
LE = r'(?P<LE><=)'
EQ = r'(?P<EQ>>=)'
master_pat = re.compile("|".join([LE, LT, EQ]))  # 正确的顺序
master_pat = re.compile("|".join([LT, LE, EQ]))  # 错误的顺序

第二种顺序的错误之处在于,这样会将'<='文本错误地匹配为LT('<')紧跟着EQ('='),而未能正确匹配为单独的LE(<=)。

对于“有可能”形成子串的模式,我们也需保持警惕,例如下面这种情况:

PRINT = r'(?P<PRINT>print)'
NAME = r'(?P<NAME>[a-zA-Z_][a-zA-Z_0-9]*)'

master_pat = re.compile("|".join([PRINT, NAME]))  # 正确的顺序

for tok in generate_tokens(master_pat, "printer"):
    print(tok)

可以看到,print实际上成为了另一个模式的子串,从而导致另一个模式的匹配出现问题:

# Token(type='PRINT', value='print')
# Token(type='NAME', value='er')

对于更高级的语法分词,建议采用如PyParsing或PLY等专业包。特别地,针对英文自然语言文章的分词,通常已集成到各类NLP包中(一般包含按空格拆分、处理前后缀、去除停用词三个步骤)。对于中文自然语言处理分词,也有丰富的工具可供选择(例如jieba分词工具包)。

到此这篇关于Python利用re模块实现简易分词(tokenization)的文章就介绍到这了,更多相关Python 分词内容请搜索本站以前的文章或继续浏览下面的相关文章,希望大家以后多多支持本站!

免责声明:本文内容仅供参考,不代表本站观点。转载或引用请注明出处。

Python 相关产品和应用

全部下载
Selenium

Selenium

4.51.0 · Windows

用于浏览器自动化和 Web 测试的 Python 工具。

scikit-learn

scikit-learn

1.9.1 · Windows

常用 Python 机器学习算法与模型评估工具库。

Black

Black

26.10.1 · Windows

强调一致性的 Python 自动代码格式化工具。

Ruff

Ruff

0.17.1 · macOS

由 Rust 编写的高速 Python 代码检查和格式化工具。

pytest

pytest

9.1.1 · Windows

简洁而强大的 Python 自动化测试框架。

FastAPI

FastAPI

0.146.0 · Windows

高性能 Python API 框架,支持类型提示和自动接口文档。

Flask

Flask

3.1.3 · Windows

轻量灵活的 Python Web 应用框架。

Django

Django

6.1.2 · Windows

成熟的 Python Web 框架,适合构建内容站点和业务系统。

Spyder

Spyder

6.1.7 · Windows

面向科学计算与数据分析的 Python 集成开发环境。

SciPy

SciPy

1.18.1 · Windows

面向科学、工程和技术计算的 Python 算法库。