前言
下面介绍如何在 Python 中基于 flashtext 模块使用 FlashText 算法进行字符串查找和替换。
1. 准备
开始之前,你需要确保 Python 和 pip 已经成功安装在电脑上。
请选择以下任一种方式输入命令安装依赖:
- Windows 环境打开 Cmd (开始 - 运行-CMD)。
- MacOS 环境打开 Terminal (command+空格输入 Terminal)。
- 如果你用的是 VSCode 编辑器 或 Pycharm,可以直接使用界面下方的 Terminal.
pip install flashtext
2. 基本使用
提取关键词
一个最基本的提取关键词的例子如下:
from flashtext import KeywordProcessor
# 1. 初始化关键字处理器
keyword_processor = KeywordProcessor()
# 2. 添加关键词
keyword_processor.add_keyword('Big Apple', 'New York')
keyword_processor.add_keyword('Bay Area')
# 3. 处理目标句子并提取相应关键词
keywords_found = keyword_processor.extract_keywords('I love Big Apple and Bay Area.')
# 4. 结果
print(keywords_found)
# ['New York', 'Bay Area']
其中 add_keyword 的第一个参数代表需要被查找的关键词,第二个参数是给这个关键词一个别名,如果找到了则以别名显示。
替换关键词
如果你想要替换关键词,只需要调用处理器的 replace_keywords 函数:
from flashtext import KeywordProcessor
# 1. 初始化关键字处理器
keyword_processor = KeywordProcessor()
# 2. 添加关键词
keyword_processor.add_keyword('New Delhi', 'NCR region')
# 3. 替换关键词
new_sentence = keyword_processor.replace_keywords('I love Big Apple and new delhi.')
# 4. 结果
print(new_sentence)
# 'I love New York and NCR region.'
关键词大小写敏感
如果你需要精确提取,识别大小写字母,那么你可以在处理器初始化的时候设定 sensitive 参数:
flashtext KeywordProcessor
keyword_processor = KeywordProcessor(case_sensitive=)
keyword_processor.add_keyword(, )
keyword_processor.add_keyword()
keywords_found = keyword_processor.extract_keywords()
(keywords_found)


