Pythonのpytrieモジュールで文字列のプレフィックス（前方一致）マッチングを行う方法

本記事では、Pythonのpytrieモジュールを使用して、文字列のリストから指定したプレフィックス（前方一致）に合致する文字列を抽出する方法を解説します。まず、具体例を見て理解を深めましょう。

入力:
リスト: ['tutorialspoint', 'tutorials', 'tutorialspython', 'python']
プレフィックス: 'tutorials'

出力:
['tutorialspoint', 'tutorials', 'tutorialspython']

この処理はさまざまな方法で実装できますが、本チュートリアルではpytrieモジュールを使ったアプローチを紹介します。

pytrieモジュールとは

pytrieモジュールが提供するpytrie.StringTrieは、「トライ（Trie）」と呼ばれる木構造のデータ構造です。これにより、作成・挿入・検索・削除といった操作を効率的に行えます。特にプレフィックス検索に強く、大量の文字列データから前方一致する要素を高速に取り出せるのが大きな特徴です。

インストール方法

まず、以下のコマンドでpytrieモジュールをインストールします。

pip install pytrie

実装の手順

目的の出力を得るための手順は次の通りです。

• pytrieモジュールをインポートする。
• 文字列リストとプレフィックスを初期化する。
• pytrie.StringTrie() でトライ構造を作成する。
• リストを反復処理しながらトライに挿入する。
• 指定したプレフィックスに合致する値を出力する。

コード例

# モジュールのインポート
import pytrie

# リストとプレフィックスの初期化
strings = ['tutorialspoint', 'tutorials', 'tutorialspython', 'python', 'learnpython']
prefix = 'tutorials'

# トライ構造を作成
trie = pytrie.StringTrie()

# リストを反復処理してトライに追加
for item in strings:
    trie[item] = item

# マッチした文字列を出力
print(trie.values(prefix))

上記のコードを実行すると、次のような結果が得られます。

実行結果

['tutorials', 'tutorialspoint', 'tutorialspython']

まとめ

pytrieモジュールのStringTreeを利用すれば、プレフィックスに一致する文字列をシンプルかつ効率的に抽出できます。辞書型のようにキーと値を扱え、values()メソッドにプレフィックスを渡すだけで前方一致検索が完了する点が魅力です。大量の文字列データを扱うケースや、検索候補の自動補完（オートコンプリート）機能などを実装したい場合に特に有効でしょう。