使用pytrie模块在Python中进行前缀匹配

在本文中,我们将学习pytrie模块,以从字符串列表中为匹配的字符串添加前缀。让我们看一个例子来清楚地理解它。

Input:
List: ['nhooo', 'tutorials', 'tutorialspython', 'python']
Prefix: 'tutorials'

Output:
['nhooo', 'tutorials', 'tutorialspython']

我们可以通过不同的方式来实现它。在本教程中,我们将使用pytrie模块来实现它。

pytrie模块中,我们将使用pytrie.StringTrie数据结构。我们可以执行创建,插入,搜索删除操作。

首先,使用以下命令安装pytrie模块。

pip install pytrie

让我们看看实现所需输出的步骤。

  • 导入pytrie模块。

  • 初始化列表,前缀。

  • 使用pytrie.StringTrie()创建一个trie数据结构。

  • 遍历列表并插入到trie结构中。

  • 并打印与给定前缀匹配的值。

示例

# importing the module
import pytrie

# initializing the list and prefix
strings = ['nhooo', 'tutorials', 'tutorialspython', 'python', 'learnpython']
prefix = 'tutorials'

# creating an trie data structure
trie = pytrie.StringTrie()

# iterating over the list and adding it to trie
for item in strings:
   trie[item] = item

# printing the matched strings
print(trie.values(prefix))

如果执行以上代码,则将得到以下结果。

输出结果

['tutorials', 'nhooo', 'tutorialspython']

结论

如果您对本教程有任何疑问,请在评论部分中提及。