pygtrie is a pure Python implementation of a trie data structure.
Trie data structure, also known as radix or prefix tree, is a tree associating keys to values where all the descendants of a node have a common prefix (associated with that node).
The trie module contains Trie, CharTrie and StringTrie classes each implementing a mutable mapping interface, i.e. dict interface. As such, in most circumstances, Trie could be used as a drop-in replacement for a dict, but the prefix nature of the data structure is trie’s real strength.
The module also contains :PrefixSet class which uses a trie to store a set of prefixes such that a key is contained in the set if it, or any of its prefixes, is stored in the set.
Features
A full mutable mapping implementation.
Supports iterating over as well as deleting a subtrie.
Supports prefix checking as well as shortest and longest prefix look-up.
Extensible for any kind of user-defined keys.
A PrefixSet supports “all keys starting with given prefix” logic.
Can store any value including None.
For example usage, see scripts in examples directory.
Installation
To install pygtrie, simply run:
pip install pygtrie
or by adding line such as:
pygtrie == 2.*
to project’s requirements file. Alternatively, if installation from source is desired, it can be achieved by executing:
python setup.py install
Version History
2.6.1: 2026/09/01
Added python_requires metadata to indicate Python 3.11 requirement. [Thanks to skshetry for reporting]
2.6: 2026/09/01 [pulled back from PyPi]
Python 3.11 is now required. Users still on 3.10 need to hold off till they upgrade their Python version (3.10 is reaching end-of-life in a couple months) or temporarily vendor the module and replace all instances of _t.Self in pygtrie.py with _t.Any.
Add type annotation to the code base. This enables better static type analysis on the code bases using pygtrie.
There are a few corner cases where the type annotations aren’t entirely sound. Most notably, the pygtrie.Trie class always returns keys as tuple[S, ...] regardless of declared type. The documentation points out ways to deal with it.
[Thanks to Dave Tapley and Avasam for requesting and discussion the feature]
Deprecated and warn about some methods of pygtrie._NoneStep returned by pygtrie.Trie.shortest_prefix and pygtrie.Trie.longest_prefix when no prefix is found.
Historically, prefixes were returned as (key, value) pairs and to maintain compatibility, lack of a prefix was signalled by (None, None) pair. However, treating lack of prefix as a tuple has long been deprecated:
>>> result = CharTrie(foo=42).longest_prefix('bar') >>> key, value = result # Currently, (None, None); >>> # in the future, will raise TypeError. >>> key = result.key # Currently None; >>> # in the future will raise AttributeError. >>> val = result.value # Currently None; >>> # in the future will raise AttributeError.Truth value testing can be used to see whether prefix exist, and pygtrie._NoneStep.get method can be used to safely get value of a prefix with a fallback if prefix isn’t valid:
>>> result = CharTrie(foo=42).longest_prefix('bar') >>> if result: ... key = result.key ... else: ... key = None >>> value = result.get(None)Behaviour when prefix exists remains unchanged:
>>> result = CharTrie(foo=42).longest_prefix('foobar') >>> key, value = result >>> assert (key, value) == ('foo', 42) >>> key, value = result.key, result.value >>> assert (key, value) == ('foo', 42)Add deprecation warning to pygtrie._Step.set method. pygtrie._Step is returned methods such as pygtrie.Trie.shortest_prefix and pygtrie.Trie.prefixes and represent a valid prefix of a key. The method has been deprecated since version 2.3.3; it’ll now issue a warning when used. Proper way to set value of a prefix is via value property, e.g.:
>>> prefix = CharTrie(foo=0, foobar=0).longest_prefix('foobarbaz') >>> prefix.value += 1Fixed pygtrie._Step string conversion raising an exception if step represents node without value. In previous versions the following would raise KeyError:
>>> list(map(repr, CharTrie(a=42).walk_towards('a'))) ["('': <no value>)", "('a': 42)"]Remove obsolete license classifiers from the package metadata. [Thanks to Benjamin T. Schwertfeger for reporting]
2.5: 2022/07/16
Add pygtrie.Trie.merge method which merges structures of two tries.
Add pygtrie.Trie.strictly_equals method which compares two tries with stricter rules than regular equality operator. It’s not sufficient that keys and values are the same but the structure of the tries must be the same as well. For example:
>>> t0 = StringTrie({'foo/bar.baz': 42}, separator='/') >>> t1 = StringTrie({'foo/bar.baz': 42}, separator='.') >>> t0 == t1 True >>> t0.strictly_equals(t1) FalseFix pygtrie.Trie.__eq__ implementation such that key values are taken into consideration rather than just looking at trie structure. To see what this means it’s best to look at a few examples. Firstly:
>>> t0 = StringTrie({'foo/bar': 42}, separator='/') >>> t1 = StringTrie({'foo.bar': 42}, separator='.') >>> t0 == t1 FalseThis used to be true since the two tries have the same node structure. However, as far as Mapping interface is concerned, they use different keys, i.e. `set(t0) != set(t1). Secondly:
>>> t0 = StringTrie({'foo/bar.baz': 42}, separator='/') >>> t1 = StringTrie({'foo/bar.baz': 42}, separator='.') >>> t0 == t1 TrueThis used to be false since the two tries have different node structures (the first one splits key into ('foo', 'bar.baz') while the second into ('foo/bar', 'baz')). However, their keys are the same, i.e. `set(t0) == set(t1). And lastly:
>>> t0 = Trie({'foo': 42}) >>> t1 = CharTrie({'foo': 42}) >>> t0 == t1 FalseThis used to be true since the two tries have the same node structure. However, the two classes return key as different values. pygtrie.Trie returns keys as tuples while pygtrie.CharTrie returns them as strings.
2.4.2: 2021/01/03
Remove use of ‘super’ in setup.py to fix compatibility with Python 2.7. This changes build code only; no changes to the library itself.
2.4.1: 2020/11/20
Remove dependency on packaging module from setup.py to fix installation on systems without that package. This changes build code only; no changes to the library itself. [Thanks to Eric McLachlan for reporting]
2.4.0: 2020/11/19 [pulled back from PyPi]
Change children argument of the node_factory passed to pygtrie.Trie.traverse from a generator to an iterator with a custom bool conversion. This allows checking whether node has children without having to iterate over them (bool(children))
To test whether this feature is available, one can check whether Trie.traverse.uses_bool_convertible_children property is true, e.g.: getattr(pygtrie.Trie.traverse, 'uses_bool_convertible_children', False).
[Thanks to Pallab Pain for suggesting the feature]
2.3.3: 2020/04/04
Fix to ‘AttributeError: _NoChildren object has no attribute sorted_items’ failure when iterating over a trie with sorting enabled. [Thanks to Pallab Pain for reporting]
Add value property setter to step objects returned by pygtrie.Trie.walk_towards et al. This deprecates the set method.
The module now exports pygtrie.__version__ making it possible to determine version of the library at run-time.
2.3.2: 2019/07/18
Trivial metadata fix
2.3.1: 2019/07/18 [pulled back from PyPi]
Fix to pygtrie.PrefixSet initialisation incorrectly storing elements even if their prefixes are also added to the set.
For example, PrefixSet(('foo', 'foobar')) incorrectly resulted in a two-element set even though the interface dictates that only foo is kept (recall that if foo is member of the set, foobar is as well). [Thanks to Tal Maimon for reporting]
Fix to pygtrie.Trie.copy method not preserving enable-sorting flag and, in case of pygtrie.StringTrie, separator property.
Add support for the copy module so copy.copy can now be used with trie objects.
Leafs and nodes with just one child use more memory-optimised representation which reduces overall memory usage of a trie structure.
Minor performance improvement for adding new elements to a pygtrie.PrefixSet.
Improvements to string representation of objects which now includes type and, for pygtrie.StringTrie object, value of separator property.
Metadata
Release files for pygtrie 2.6.1
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| pygtrie-2.6.1.tar.gz | 56.3 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| pygtrie-2.6.1-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 87.4 kB
Release files / pygtrie-2.6.1.tar.gz
| Download URL | pygtrie-2.6.1.tar.gz |
|---|---|
| Size | 56.3 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
1934613126070b4ec51a3cefcf0c94cebb0139043b444c0f02486caebd0d8011
|
|
BLAKE2b-256 checksum How to use checksums |
b113275484b51d6bcb18b372b65ece2a238e5e3b778f1b7b10b0930408819a88
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/6.2.0 CPython/3.13.15
|
Release files / pygtrie-2.6.1-py3-none-any.whl
| Download URL | pygtrie-2.6.1-py3-none-any.whl |
|---|---|
| Size | 31.1 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
0ad593c596f938f0f468e05c5af174f25a7237b14fae6674843c60d63ae4313c
|
|
BLAKE2b-256 checksum How to use checksums |
f5a17dc65e8997fa3d46f52b94f753344a2c3d58f7fa2722f8827a8291f7f075
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/6.2.0 CPython/3.13.15
|