blob: ca2f1167cc86d4cfe2eb96424ff2ce0fecdc9cda [file] [log] [blame]
Raymond Hettinger53dbe392008-02-12 20:03:09 +00001:mod:`collections` --- Container datatypes
2==========================================
Georg Brandl116aa622007-08-15 14:28:22 +00003
4.. module:: collections
Raymond Hettinger7929cfb2012-06-09 19:15:26 -07005 :synopsis: Container datatypes
Terry Jan Reedyfa089b92016-06-11 15:02:54 -04006
Georg Brandl116aa622007-08-15 14:28:22 +00007.. moduleauthor:: Raymond Hettinger <python@rcn.com>
8.. sectionauthor:: Raymond Hettinger <python@rcn.com>
9
Terry Jan Reedyfa089b92016-06-11 15:02:54 -040010**Source code:** :source:`Lib/collections/__init__.py`
11
Christian Heimesfe337bf2008-03-23 21:54:12 +000012.. testsetup:: *
13
Raymond Hettinger7929cfb2012-06-09 19:15:26 -070014 from collections import *
15 import itertools
16 __name__ = '<doctest>'
Georg Brandl116aa622007-08-15 14:28:22 +000017
Raymond Hettinger4f707fd2011-01-10 19:54:11 +000018--------------
19
Raymond Hettingera6b76ba2010-08-08 00:29:08 +000020This module implements specialized container datatypes providing alternatives to
21Python's general purpose built-in containers, :class:`dict`, :class:`list`,
22:class:`set`, and :class:`tuple`.
Christian Heimes0bd4e112008-02-12 22:59:25 +000023
Raymond Hettingera6b76ba2010-08-08 00:29:08 +000024===================== ====================================================================
25:func:`namedtuple` factory function for creating tuple subclasses with named fields
26:class:`deque` list-like container with fast appends and pops on either end
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +000027:class:`ChainMap` dict-like class for creating a single view of multiple mappings
Raymond Hettingera6b76ba2010-08-08 00:29:08 +000028:class:`Counter` dict subclass for counting hashable objects
29:class:`OrderedDict` dict subclass that remembers the order entries were added
30:class:`defaultdict` dict subclass that calls a factory function to supply missing values
31:class:`UserDict` wrapper around dictionary objects for easier dict subclassing
32:class:`UserList` wrapper around list objects for easier list subclassing
33:class:`UserString` wrapper around string objects for easier string subclassing
34===================== ====================================================================
Georg Brandl116aa622007-08-15 14:28:22 +000035
Raymond Hettinger158c9c22011-02-22 00:41:50 +000036.. versionchanged:: 3.3
Raymond Hettinger7929cfb2012-06-09 19:15:26 -070037 Moved :ref:`collections-abstract-base-classes` to the :mod:`collections.abc` module.
Raymond Hettingere6d34212018-01-29 08:27:49 -080038 For backwards compatibility, they continue to be visible in this module through
39 Python 3.7. Subsequently, they will be removed entirely.
Mark Summerfield08898b42007-09-05 08:43:04 +000040
41
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +000042:class:`ChainMap` objects
43-------------------------
44
Georg Brandl283b96b2012-04-03 09:16:46 +020045.. versionadded:: 3.3
46
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +000047A :class:`ChainMap` class is provided for quickly linking a number of mappings
48so they can be treated as a single unit. It is often much faster than creating
49a new dictionary and running multiple :meth:`~dict.update` calls.
50
51The class can be used to simulate nested scopes and is useful in templating.
52
53.. class:: ChainMap(*maps)
54
Raymond Hettinger7929cfb2012-06-09 19:15:26 -070055 A :class:`ChainMap` groups multiple dicts or other mappings together to
56 create a single, updateable view. If no *maps* are specified, a single empty
57 dictionary is provided so that a new chain always has at least one mapping.
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +000058
Raymond Hettinger7929cfb2012-06-09 19:15:26 -070059 The underlying mappings are stored in a list. That list is public and can
Martin Panter8d56c022016-05-29 04:13:35 +000060 be accessed or updated using the *maps* attribute. There is no other state.
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +000061
Raymond Hettinger7929cfb2012-06-09 19:15:26 -070062 Lookups search the underlying mappings successively until a key is found. In
63 contrast, writes, updates, and deletions only operate on the first mapping.
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +000064
Raymond Hettinger7929cfb2012-06-09 19:15:26 -070065 A :class:`ChainMap` incorporates the underlying mappings by reference. So, if
66 one of the underlying mappings gets updated, those changes will be reflected
67 in :class:`ChainMap`.
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +000068
Raymond Hettinger7929cfb2012-06-09 19:15:26 -070069 All of the usual dictionary methods are supported. In addition, there is a
70 *maps* attribute, a method for creating new subcontexts, and a property for
71 accessing all but the first mapping:
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +000072
Raymond Hettinger7929cfb2012-06-09 19:15:26 -070073 .. attribute:: maps
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +000074
Raymond Hettinger7929cfb2012-06-09 19:15:26 -070075 A user updateable list of mappings. The list is ordered from
76 first-searched to last-searched. It is the only stored state and can
77 be modified to change which mappings are searched. The list should
78 always contain at least one mapping.
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +000079
Vinay Sajip1ba81ee2013-01-11 23:39:53 +000080 .. method:: new_child(m=None)
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +000081
Vinay Sajip1ba81ee2013-01-11 23:39:53 +000082 Returns a new :class:`ChainMap` containing a new map followed by
83 all of the maps in the current instance. If ``m`` is specified,
84 it becomes the new map at the front of the list of mappings; if not
85 specified, an empty dict is used, so that a call to ``d.new_child()``
86 is equivalent to: ``ChainMap({}, *d.maps)``. This method is used for
Raymond Hettinger7929cfb2012-06-09 19:15:26 -070087 creating subcontexts that can be updated without altering values in any
88 of the parent mappings.
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +000089
Vinay Sajip1ba81ee2013-01-11 23:39:53 +000090 .. versionchanged:: 3.4
91 The optional ``m`` parameter was added.
92
Raymond Hettinger2a61c452012-07-15 22:37:20 -070093 .. attribute:: parents
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +000094
Raymond Hettingerb22ba042012-07-16 02:07:41 -070095 Property returning a new :class:`ChainMap` containing all of the maps in
Raymond Hettinger2a61c452012-07-15 22:37:20 -070096 the current instance except the first one. This is useful for skipping
97 the first map in the search. Use cases are similar to those for the
98 :keyword:`nonlocal` keyword used in :term:`nested scopes <nested
99 scope>`. The use cases also parallel those for the built-in
100 :func:`super` function. A reference to ``d.parents`` is equivalent to:
101 ``ChainMap(*d.maps[1:])``.
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +0000102
Raymond Hettinger86f093f2019-02-21 09:12:15 -0800103 Note, the iteration order of a :class:`ChainMap()` is determined by
104 scanning the mappings last to first::
105
106 >>> baseline = {'music': 'bach', 'art': 'rembrandt'}
107 >>> adjustments = {'art': 'van gogh', 'opera': 'carmen'}
108 >>> list(ChainMap(adjustments, baseline))
109 ['music', 'art', 'opera']
110
111 This gives the same ordering as a series of :meth:`dict.update` calls
112 starting with the last mapping::
113
114 >>> combined = baseline.copy()
115 >>> combined.update(adjustments)
116 >>> list(combined)
117 ['music', 'art', 'opera']
Raymond Hettingerbfcb4292012-06-10 11:39:44 -0700118
119.. seealso::
120
121 * The `MultiContext class
Sandro Tosiea475302012-08-12 10:37:23 +0200122 <https://github.com/enthought/codetools/blob/4.0.0/codetools/contexts/multi_context.py>`_
Raymond Hettingerbfcb4292012-06-10 11:39:44 -0700123 in the Enthought `CodeTools package
124 <https://github.com/enthought/codetools>`_ has options to support
125 writing to any mapping in the chain.
126
127 * Django's `Context class
Georg Brandl525d3552014-10-29 10:26:56 +0100128 <https://github.com/django/django/blob/master/django/template/context.py>`_
Raymond Hettingerbfcb4292012-06-10 11:39:44 -0700129 for templating is a read-only chain of mappings. It also features
130 pushing and popping of contexts similar to the
131 :meth:`~collections.ChainMap.new_child` method and the
Michael Seiferte1052942018-03-26 13:40:35 +0200132 :attr:`~collections.ChainMap.parents` property.
Raymond Hettingerbfcb4292012-06-10 11:39:44 -0700133
134 * The `Nested Contexts recipe
Serhiy Storchaka6dff0202016-05-07 10:49:07 +0300135 <https://code.activestate.com/recipes/577434/>`_ has options to control
Raymond Hettingerbfcb4292012-06-10 11:39:44 -0700136 whether writes and other mutations apply only to the first mapping or to
137 any mapping in the chain.
138
139 * A `greatly simplified read-only version of Chainmap
Serhiy Storchaka6dff0202016-05-07 10:49:07 +0300140 <https://code.activestate.com/recipes/305268/>`_.
Raymond Hettingerbfcb4292012-06-10 11:39:44 -0700141
142
143:class:`ChainMap` Examples and Recipes
144^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
145
146This section shows various approaches to working with chained maps.
147
148
149Example of simulating Python's internal lookup chain::
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +0000150
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700151 import builtins
152 pylookup = ChainMap(locals(), globals(), vars(builtins))
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +0000153
Raymond Hettingerb2269ba2012-07-15 23:53:32 -0700154Example of letting user specified command-line arguments take precedence over
155environment variables which in turn take precedence over default values::
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +0000156
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700157 import os, argparse
Raymond Hettingerb2269ba2012-07-15 23:53:32 -0700158
159 defaults = {'color': 'red', 'user': 'guest'}
160
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700161 parser = argparse.ArgumentParser()
162 parser.add_argument('-u', '--user')
163 parser.add_argument('-c', '--color')
Raymond Hettingerb2269ba2012-07-15 23:53:32 -0700164 namespace = parser.parse_args()
165 command_line_args = {k:v for k, v in vars(namespace).items() if v}
166
167 combined = ChainMap(command_line_args, os.environ, defaults)
168 print(combined['color'])
169 print(combined['user'])
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +0000170
Raymond Hettingerbfcb4292012-06-10 11:39:44 -0700171Example patterns for using the :class:`ChainMap` class to simulate nested
172contexts::
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +0000173
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700174 c = ChainMap() # Create root context
175 d = c.new_child() # Create nested child context
176 e = c.new_child() # Child of c, independent from d
177 e.maps[0] # Current context dictionary -- like Python's locals()
178 e.maps[-1] # Root context -- like Python's globals()
179 e.parents # Enclosing context chain -- like Python's nonlocals
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +0000180
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700181 d['x'] = 1 # Set value in current context
Raymond Hettinger9b0c6812019-02-20 13:49:23 -0800182 d['x'] # Get first key in the chain of contexts
Andrew Svetlov1a8db9c2012-10-04 19:29:25 +0300183 del d['x'] # Delete from current context
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700184 list(d) # All nested values
185 k in d # Check all nested values
186 len(d) # Number of nested values
187 d.items() # All nested items
188 dict(d) # Flatten into a regular dictionary
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +0000189
Raymond Hettingerbfcb4292012-06-10 11:39:44 -0700190The :class:`ChainMap` class only makes updates (writes and deletions) to the
191first mapping in the chain while lookups will search the full chain. However,
192if deep writes and deletions are desired, it is easy to make a subclass that
193updates keys found deeper in the chain::
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +0000194
Raymond Hettingerbfcb4292012-06-10 11:39:44 -0700195 class DeepChainMap(ChainMap):
196 'Variant of ChainMap that allows direct updates to inner scopes'
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +0000197
Raymond Hettingerbfcb4292012-06-10 11:39:44 -0700198 def __setitem__(self, key, value):
199 for mapping in self.maps:
200 if key in mapping:
201 mapping[key] = value
202 return
203 self.maps[0][key] = value
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +0000204
Raymond Hettingerbfcb4292012-06-10 11:39:44 -0700205 def __delitem__(self, key):
206 for mapping in self.maps:
207 if key in mapping:
208 del mapping[key]
209 return
210 raise KeyError(key)
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +0000211
Serhiy Storchakaf47036c2013-12-24 11:04:36 +0200212 >>> d = DeepChainMap({'zebra': 'black'}, {'elephant': 'blue'}, {'lion': 'yellow'})
Raymond Hettingerbfcb4292012-06-10 11:39:44 -0700213 >>> d['lion'] = 'orange' # update an existing key two levels down
214 >>> d['snake'] = 'red' # new keys get added to the topmost dict
215 >>> del d['elephant'] # remove an existing key one level down
Raymond Hettinger2064bb62018-09-10 18:43:08 -0700216 >>> d # display result
Raymond Hettingerbfcb4292012-06-10 11:39:44 -0700217 DeepChainMap({'zebra': 'black', 'snake': 'red'}, {}, {'lion': 'orange'})
Georg Brandl4dcf4742012-03-08 20:35:08 +0100218
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +0000219
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000220:class:`Counter` objects
221------------------------
222
223A counter tool is provided to support convenient and rapid tallies.
224For example::
225
Raymond Hettinger1c62dc92009-02-04 11:41:45 +0000226 >>> # Tally occurrences of words in a list
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000227 >>> cnt = Counter()
Raymond Hettinger670eaec2009-01-21 23:14:07 +0000228 >>> for word in ['red', 'blue', 'red', 'green', 'blue', 'blue']:
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000229 ... cnt[word] += 1
230 >>> cnt
231 Counter({'blue': 3, 'red': 2, 'green': 1})
232
Raymond Hettinger1c62dc92009-02-04 11:41:45 +0000233 >>> # Find the ten most common words in Hamlet
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000234 >>> import re
Raymond Hettingerfaaba592013-03-01 03:30:20 -0800235 >>> words = re.findall(r'\w+', open('hamlet.txt').read().lower())
Raymond Hettinger0bae6622009-01-20 13:00:59 +0000236 >>> Counter(words).most_common(10)
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000237 [('the', 1143), ('and', 966), ('to', 762), ('of', 669), ('i', 631),
238 ('you', 554), ('a', 546), ('my', 514), ('hamlet', 471), ('in', 451)]
239
240.. class:: Counter([iterable-or-mapping])
241
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700242 A :class:`Counter` is a :class:`dict` subclass for counting hashable objects.
Andrés Delfino76e8fd72018-11-11 13:56:47 -0300243 It is a collection where elements are stored as dictionary keys
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700244 and their counts are stored as dictionary values. Counts are allowed to be
245 any integer value including zero or negative counts. The :class:`Counter`
246 class is similar to bags or multisets in other languages.
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000247
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700248 Elements are counted from an *iterable* or initialized from another
249 *mapping* (or counter):
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000250
Raymond Hettinger73662a52009-01-27 02:38:22 +0000251 >>> c = Counter() # a new, empty counter
252 >>> c = Counter('gallahad') # a new counter from an iterable
253 >>> c = Counter({'red': 4, 'blue': 2}) # a new counter from a mapping
254 >>> c = Counter(cats=4, dogs=8) # a new counter from keyword args
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000255
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700256 Counter objects have a dictionary interface except that they return a zero
257 count for missing items instead of raising a :exc:`KeyError`:
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000258
Raymond Hettinger94adc8e2009-01-22 05:27:37 +0000259 >>> c = Counter(['eggs', 'ham'])
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000260 >>> c['bacon'] # count of a missing element is zero
261 0
262
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700263 Setting a count to zero does not remove an element from a counter.
264 Use ``del`` to remove it entirely:
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000265
Raymond Hettinger94adc8e2009-01-22 05:27:37 +0000266 >>> c['sausage'] = 0 # counter entry with a zero count
267 >>> del c['sausage'] # del actually removes the entry
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000268
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700269 .. versionadded:: 3.1
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000270
271
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700272 Counter objects support three methods beyond those available for all
273 dictionaries:
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000274
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700275 .. method:: elements()
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000276
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700277 Return an iterator over elements repeating each as many times as its
278 count. Elements are returned in arbitrary order. If an element's count
279 is less than one, :meth:`elements` will ignore it.
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000280
Raymond Hettinger0bae6622009-01-20 13:00:59 +0000281 >>> c = Counter(a=4, b=2, c=0, d=-2)
Zachary Ware2b52c0a2016-08-09 17:38:22 -0500282 >>> sorted(c.elements())
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000283 ['a', 'a', 'a', 'a', 'b', 'b']
284
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700285 .. method:: most_common([n])
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000286
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700287 Return a list of the *n* most common elements and their counts from the
Raymond Hettingera3780252015-05-13 02:47:57 -0700288 most common to the least. If *n* is omitted or ``None``,
Michael Seiferte1052942018-03-26 13:40:35 +0200289 :meth:`most_common` returns *all* elements in the counter.
Raymond Hettinger3afdb282015-05-13 14:39:04 -0700290 Elements with equal counts are ordered arbitrarily:
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000291
Zachary Ware2b52c0a2016-08-09 17:38:22 -0500292 >>> Counter('abracadabra').most_common(3) # doctest: +SKIP
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000293 [('a', 5), ('r', 2), ('b', 2)]
294
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700295 .. method:: subtract([iterable-or-mapping])
Raymond Hettinger9c01e442010-04-03 10:32:58 +0000296
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700297 Elements are subtracted from an *iterable* or from another *mapping*
298 (or counter). Like :meth:`dict.update` but subtracts counts instead
299 of replacing them. Both inputs and outputs may be zero or negative.
Raymond Hettinger9c01e442010-04-03 10:32:58 +0000300
301 >>> c = Counter(a=4, b=2, c=0, d=-2)
302 >>> d = Counter(a=1, b=2, c=3, d=4)
303 >>> c.subtract(d)
Andrew Svetlovf6351722012-12-17 14:01:16 +0200304 >>> c
Raymond Hettinger9c01e442010-04-03 10:32:58 +0000305 Counter({'a': 3, 'b': 0, 'c': -3, 'd': -6})
306
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700307 .. versionadded:: 3.2
Ezio Melotti0be8b1c2010-04-04 06:53:44 +0000308
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700309 The usual dictionary methods are available for :class:`Counter` objects
310 except for two which work differently for counters.
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000311
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700312 .. method:: fromkeys(iterable)
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000313
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700314 This class method is not implemented for :class:`Counter` objects.
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000315
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700316 .. method:: update([iterable-or-mapping])
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000317
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700318 Elements are counted from an *iterable* or added-in from another
319 *mapping* (or counter). Like :meth:`dict.update` but adds counts
320 instead of replacing them. Also, the *iterable* is expected to be a
321 sequence of elements, not a sequence of ``(key, value)`` pairs.
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000322
323Common patterns for working with :class:`Counter` objects::
324
Raymond Hettinger73662a52009-01-27 02:38:22 +0000325 sum(c.values()) # total of all counts
326 c.clear() # reset all counts
327 list(c) # list unique elements
328 set(c) # convert to a set
329 dict(c) # convert to a regular dictionary
330 c.items() # convert to a list of (elem, cnt) pairs
331 Counter(dict(list_of_pairs)) # convert from a list of (elem, cnt) pairs
Georg Brandl87f3d7b2013-10-06 12:36:39 +0200332 c.most_common()[:-n-1:-1] # n least common elements
Raymond Hettingerfcb393c2011-08-09 13:00:40 -0700333 +c # remove zero and negative counts
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000334
Raymond Hettinger72a95cc2009-02-25 22:51:40 +0000335Several mathematical operations are provided for combining :class:`Counter`
336objects to produce multisets (counters that have counts greater than zero).
337Addition and subtraction combine counters by adding or subtracting the counts
338of corresponding elements. Intersection and union return the minimum and
339maximum of corresponding counts. Each operation can accept inputs with signed
340counts, but the output will exclude results with counts of zero or less.
Raymond Hettinger4d2073a2009-01-20 03:41:22 +0000341
Raymond Hettingere0d1b9f2009-01-21 20:36:27 +0000342 >>> c = Counter(a=3, b=1)
343 >>> d = Counter(a=1, b=2)
Raymond Hettinger73662a52009-01-27 02:38:22 +0000344 >>> c + d # add two counters together: c[x] + d[x]
Raymond Hettinger4d2073a2009-01-20 03:41:22 +0000345 Counter({'a': 4, 'b': 3})
Raymond Hettinger73662a52009-01-27 02:38:22 +0000346 >>> c - d # subtract (keeping only positive counts)
Raymond Hettinger4d2073a2009-01-20 03:41:22 +0000347 Counter({'a': 2})
Zachary Ware2b52c0a2016-08-09 17:38:22 -0500348 >>> c & d # intersection: min(c[x], d[x]) # doctest: +SKIP
Raymond Hettinger4d2073a2009-01-20 03:41:22 +0000349 Counter({'a': 1, 'b': 1})
Raymond Hettinger73662a52009-01-27 02:38:22 +0000350 >>> c | d # union: max(c[x], d[x])
Raymond Hettinger4d2073a2009-01-20 03:41:22 +0000351 Counter({'a': 3, 'b': 2})
352
Berker Peksag315e1042015-05-19 01:36:55 +0300353Unary addition and subtraction are shortcuts for adding an empty counter
Raymond Hettingerfcb393c2011-08-09 13:00:40 -0700354or subtracting from an empty counter.
355
356 >>> c = Counter(a=2, b=-4)
357 >>> +c
358 Counter({'a': 2})
359 >>> -c
360 Counter({'b': 4})
361
362.. versionadded:: 3.3
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700363 Added support for unary plus, unary minus, and in-place multiset operations.
Raymond Hettingerfcb393c2011-08-09 13:00:40 -0700364
Raymond Hettinger22f18852010-04-12 21:45:14 +0000365.. note::
366
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700367 Counters were primarily designed to work with positive integers to represent
368 running counts; however, care was taken to not unnecessarily preclude use
369 cases needing other types or negative values. To help with those use cases,
370 this section documents the minimum range and type restrictions.
Raymond Hettinger22f18852010-04-12 21:45:14 +0000371
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700372 * The :class:`Counter` class itself is a dictionary subclass with no
Georg Brandl2fdc0f82012-10-06 22:38:20 +0200373 restrictions on its keys and values. The values are intended to be numbers
374 representing counts, but you *could* store anything in the value field.
Raymond Hettinger22f18852010-04-12 21:45:14 +0000375
Michael Seiferte1052942018-03-26 13:40:35 +0200376 * The :meth:`~Counter.most_common` method requires only that the values be orderable.
Raymond Hettinger22f18852010-04-12 21:45:14 +0000377
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700378 * For in-place operations such as ``c[key] += 1``, the value type need only
Georg Brandl2fdc0f82012-10-06 22:38:20 +0200379 support addition and subtraction. So fractions, floats, and decimals would
380 work and negative values are supported. The same is also true for
Michael Seiferte1052942018-03-26 13:40:35 +0200381 :meth:`~Counter.update` and :meth:`~Counter.subtract` which allow negative and zero values
Georg Brandl2fdc0f82012-10-06 22:38:20 +0200382 for both inputs and outputs.
Raymond Hettinger22f18852010-04-12 21:45:14 +0000383
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700384 * The multiset methods are designed only for use cases with positive values.
Georg Brandl2fdc0f82012-10-06 22:38:20 +0200385 The inputs may be negative or zero, but only outputs with positive values
386 are created. There are no type restrictions, but the value type needs to
387 support addition, subtraction, and comparison.
Raymond Hettinger22f18852010-04-12 21:45:14 +0000388
Michael Seiferte1052942018-03-26 13:40:35 +0200389 * The :meth:`~Counter.elements` method requires integer counts. It ignores zero and
Georg Brandl2fdc0f82012-10-06 22:38:20 +0200390 negative counts.
Raymond Hettinger22f18852010-04-12 21:45:14 +0000391
Raymond Hettingerb14043c2009-01-20 23:44:31 +0000392.. seealso::
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000393
Georg Brandl5d941342016-02-26 19:37:12 +0100394 * `Bag class <https://www.gnu.org/software/smalltalk/manual-base/html_node/Bag.html>`_
Raymond Hettingerbfcb4292012-06-10 11:39:44 -0700395 in Smalltalk.
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000396
Georg Brandl5d941342016-02-26 19:37:12 +0100397 * Wikipedia entry for `Multisets <https://en.wikipedia.org/wiki/Multiset>`_.
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000398
Georg Brandl5d941342016-02-26 19:37:12 +0100399 * `C++ multisets <http://www.java2s.com/Tutorial/Cpp/0380__set-multiset/Catalog0380__set-multiset.htm>`_
Raymond Hettingerbfcb4292012-06-10 11:39:44 -0700400 tutorial with examples.
Raymond Hettingerb14043c2009-01-20 23:44:31 +0000401
Raymond Hettinger94adc8e2009-01-22 05:27:37 +0000402 * For mathematical operations on multisets and their use cases, see
Raymond Hettingerbfcb4292012-06-10 11:39:44 -0700403 *Knuth, Donald. The Art of Computer Programming Volume II,
404 Section 4.6.3, Exercise 19*.
Raymond Hettingerb14043c2009-01-20 23:44:31 +0000405
Raymond Hettinger670eaec2009-01-21 23:14:07 +0000406 * To enumerate all distinct multisets of a given size over a given set of
Michael Seiferte1052942018-03-26 13:40:35 +0200407 elements, see :func:`itertools.combinations_with_replacement`::
Raymond Hettingerb14043c2009-01-20 23:44:31 +0000408
Michael Seiferte1052942018-03-26 13:40:35 +0200409 map(Counter, combinations_with_replacement('ABC', 2)) # --> AA AB AC BB BC CC
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000410
411
Georg Brandl116aa622007-08-15 14:28:22 +0000412:class:`deque` objects
413----------------------
414
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000415.. class:: deque([iterable, [maxlen]])
Georg Brandl116aa622007-08-15 14:28:22 +0000416
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700417 Returns a new deque object initialized left-to-right (using :meth:`append`) with
418 data from *iterable*. If *iterable* is not specified, the new deque is empty.
Georg Brandl116aa622007-08-15 14:28:22 +0000419
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700420 Deques are a generalization of stacks and queues (the name is pronounced "deck"
421 and is short for "double-ended queue"). Deques support thread-safe, memory
422 efficient appends and pops from either side of the deque with approximately the
423 same O(1) performance in either direction.
Georg Brandl116aa622007-08-15 14:28:22 +0000424
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700425 Though :class:`list` objects support similar operations, they are optimized for
426 fast fixed-length operations and incur O(n) memory movement costs for
427 ``pop(0)`` and ``insert(0, v)`` operations which change both the size and
428 position of the underlying data representation.
Georg Brandl116aa622007-08-15 14:28:22 +0000429
Georg Brandl116aa622007-08-15 14:28:22 +0000430
Serhiy Storchakaecf41da2016-10-19 16:29:26 +0300431 If *maxlen* is not specified or is ``None``, deques may grow to an
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700432 arbitrary length. Otherwise, the deque is bounded to the specified maximum
433 length. Once a bounded length deque is full, when new items are added, a
434 corresponding number of items are discarded from the opposite end. Bounded
435 length deques provide functionality similar to the ``tail`` filter in
436 Unix. They are also useful for tracking transactions and other pools of data
437 where only the most recent activity is of interest.
Georg Brandl9afde1c2007-11-01 20:32:30 +0000438
Georg Brandl9afde1c2007-11-01 20:32:30 +0000439
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700440 Deque objects support the following methods:
Georg Brandl116aa622007-08-15 14:28:22 +0000441
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700442 .. method:: append(x)
Georg Brandl116aa622007-08-15 14:28:22 +0000443
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700444 Add *x* to the right side of the deque.
Georg Brandl116aa622007-08-15 14:28:22 +0000445
446
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700447 .. method:: appendleft(x)
Georg Brandl116aa622007-08-15 14:28:22 +0000448
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700449 Add *x* to the left side of the deque.
Georg Brandl116aa622007-08-15 14:28:22 +0000450
451
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700452 .. method:: clear()
Georg Brandl116aa622007-08-15 14:28:22 +0000453
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700454 Remove all elements from the deque leaving it with length 0.
Georg Brandl116aa622007-08-15 14:28:22 +0000455
456
Raymond Hettinger32ea1652015-03-21 01:37:37 -0700457 .. method:: copy()
458
459 Create a shallow copy of the deque.
460
461 .. versionadded:: 3.5
462
463
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700464 .. method:: count(x)
Raymond Hettinger44459de2010-04-03 23:20:46 +0000465
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700466 Count the number of deque elements equal to *x*.
Raymond Hettinger44459de2010-04-03 23:20:46 +0000467
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700468 .. versionadded:: 3.2
Raymond Hettinger44459de2010-04-03 23:20:46 +0000469
Georg Brandl67b21b72010-08-17 15:07:14 +0000470
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700471 .. method:: extend(iterable)
Georg Brandl116aa622007-08-15 14:28:22 +0000472
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700473 Extend the right side of the deque by appending elements from the iterable
474 argument.
Georg Brandl116aa622007-08-15 14:28:22 +0000475
476
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700477 .. method:: extendleft(iterable)
Georg Brandl116aa622007-08-15 14:28:22 +0000478
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700479 Extend the left side of the deque by appending elements from *iterable*.
480 Note, the series of left appends results in reversing the order of
481 elements in the iterable argument.
Georg Brandl116aa622007-08-15 14:28:22 +0000482
483
Raymond Hettinger855482e2015-05-23 08:57:58 -0700484 .. method:: index(x[, start[, stop]])
Raymond Hettinger32ea1652015-03-21 01:37:37 -0700485
Raymond Hettinger855482e2015-05-23 08:57:58 -0700486 Return the position of *x* in the deque (at or after index *start*
487 and before index *stop*). Returns the first match or raises
488 :exc:`ValueError` if not found.
Raymond Hettinger32ea1652015-03-21 01:37:37 -0700489
490 .. versionadded:: 3.5
491
492
493 .. method:: insert(i, x)
494
495 Insert *x* into the deque at position *i*.
496
Raymond Hettingera6389712016-02-01 21:21:19 -0800497 If the insertion would cause a bounded deque to grow beyond *maxlen*,
498 an :exc:`IndexError` is raised.
Raymond Hettinger37434322016-01-26 21:44:16 -0800499
Raymond Hettinger32ea1652015-03-21 01:37:37 -0700500 .. versionadded:: 3.5
501
502
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700503 .. method:: pop()
Georg Brandl116aa622007-08-15 14:28:22 +0000504
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700505 Remove and return an element from the right side of the deque. If no
506 elements are present, raises an :exc:`IndexError`.
Georg Brandl116aa622007-08-15 14:28:22 +0000507
508
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700509 .. method:: popleft()
Georg Brandl116aa622007-08-15 14:28:22 +0000510
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700511 Remove and return an element from the left side of the deque. If no
512 elements are present, raises an :exc:`IndexError`.
Georg Brandl116aa622007-08-15 14:28:22 +0000513
514
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700515 .. method:: remove(value)
Georg Brandl116aa622007-08-15 14:28:22 +0000516
Raymond Hettinger855482e2015-05-23 08:57:58 -0700517 Remove the first occurrence of *value*. If not found, raises a
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700518 :exc:`ValueError`.
Georg Brandl116aa622007-08-15 14:28:22 +0000519
Georg Brandl67b21b72010-08-17 15:07:14 +0000520
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700521 .. method:: reverse()
Raymond Hettingere5fdedb2009-12-10 00:47:21 +0000522
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700523 Reverse the elements of the deque in-place and then return ``None``.
Raymond Hettingere5fdedb2009-12-10 00:47:21 +0000524
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700525 .. versionadded:: 3.2
Georg Brandl116aa622007-08-15 14:28:22 +0000526
Georg Brandl67b21b72010-08-17 15:07:14 +0000527
Raymond Hettinger589c7182018-02-03 08:46:28 -0800528 .. method:: rotate(n=1)
Georg Brandl116aa622007-08-15 14:28:22 +0000529
Raymond Hettinger589c7182018-02-03 08:46:28 -0800530 Rotate the deque *n* steps to the right. If *n* is negative, rotate
531 to the left.
532
Raymond Hettingerca6c1252018-02-04 09:15:01 -0800533 When the deque is not empty, rotating one step to the right is equivalent
Raymond Hettinger589c7182018-02-03 08:46:28 -0800534 to ``d.appendleft(d.pop())``, and rotating one step to the left is
535 equivalent to ``d.append(d.popleft())``.
Benjamin Petersone41251e2008-04-25 01:59:09 +0000536
Georg Brandl116aa622007-08-15 14:28:22 +0000537
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700538 Deque objects also provide one read-only attribute:
Raymond Hettinger5bb0f0e2009-03-10 12:56:32 +0000539
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700540 .. attribute:: maxlen
Raymond Hettinger5bb0f0e2009-03-10 12:56:32 +0000541
Serhiy Storchakaecf41da2016-10-19 16:29:26 +0300542 Maximum size of a deque or ``None`` if unbounded.
Raymond Hettinger5bb0f0e2009-03-10 12:56:32 +0000543
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700544 .. versionadded:: 3.1
Raymond Hettinger5bb0f0e2009-03-10 12:56:32 +0000545
546
Georg Brandl116aa622007-08-15 14:28:22 +0000547In addition to the above, deques support iteration, pickling, ``len(d)``,
548``reversed(d)``, ``copy.copy(d)``, ``copy.deepcopy(d)``, membership testing with
Windson yang98b85352018-11-05 06:34:22 +0800549the :keyword:`in` operator, and subscript references such as ``d[0]`` to access
550the first element. Indexed access is O(1) at both ends but slows to O(n) in
551the middle. For fast random access, use lists instead.
Georg Brandl116aa622007-08-15 14:28:22 +0000552
Raymond Hettinger41290a62015-03-31 08:12:23 -0700553Starting in version 3.5, deques support ``__add__()``, ``__mul__()``,
554and ``__imul__()``.
555
Christian Heimesfe337bf2008-03-23 21:54:12 +0000556Example:
557
558.. doctest::
Georg Brandl116aa622007-08-15 14:28:22 +0000559
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700560 >>> from collections import deque
561 >>> d = deque('ghi') # make a new deque with three items
562 >>> for elem in d: # iterate over the deque's elements
563 ... print(elem.upper())
564 G
565 H
566 I
Georg Brandl116aa622007-08-15 14:28:22 +0000567
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700568 >>> d.append('j') # add a new entry to the right side
569 >>> d.appendleft('f') # add a new entry to the left side
570 >>> d # show the representation of the deque
571 deque(['f', 'g', 'h', 'i', 'j'])
Georg Brandl116aa622007-08-15 14:28:22 +0000572
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700573 >>> d.pop() # return and remove the rightmost item
574 'j'
575 >>> d.popleft() # return and remove the leftmost item
576 'f'
577 >>> list(d) # list the contents of the deque
578 ['g', 'h', 'i']
579 >>> d[0] # peek at leftmost item
580 'g'
581 >>> d[-1] # peek at rightmost item
582 'i'
Georg Brandl116aa622007-08-15 14:28:22 +0000583
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700584 >>> list(reversed(d)) # list the contents of a deque in reverse
585 ['i', 'h', 'g']
586 >>> 'h' in d # search the deque
587 True
588 >>> d.extend('jkl') # add multiple elements at once
589 >>> d
590 deque(['g', 'h', 'i', 'j', 'k', 'l'])
591 >>> d.rotate(1) # right rotation
592 >>> d
593 deque(['l', 'g', 'h', 'i', 'j', 'k'])
594 >>> d.rotate(-1) # left rotation
595 >>> d
596 deque(['g', 'h', 'i', 'j', 'k', 'l'])
Georg Brandl116aa622007-08-15 14:28:22 +0000597
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700598 >>> deque(reversed(d)) # make a new deque in reverse order
599 deque(['l', 'k', 'j', 'i', 'h', 'g'])
600 >>> d.clear() # empty the deque
601 >>> d.pop() # cannot pop from an empty deque
602 Traceback (most recent call last):
603 File "<pyshell#6>", line 1, in -toplevel-
604 d.pop()
605 IndexError: pop from an empty deque
Georg Brandl116aa622007-08-15 14:28:22 +0000606
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700607 >>> d.extendleft('abc') # extendleft() reverses the input order
608 >>> d
609 deque(['c', 'b', 'a'])
Georg Brandl116aa622007-08-15 14:28:22 +0000610
611
Georg Brandl9afde1c2007-11-01 20:32:30 +0000612:class:`deque` Recipes
613^^^^^^^^^^^^^^^^^^^^^^
Georg Brandl116aa622007-08-15 14:28:22 +0000614
615This section shows various approaches to working with deques.
616
Raymond Hettingerd2ee64d2009-03-31 22:52:48 +0000617Bounded length deques provide functionality similar to the ``tail`` filter
618in Unix::
619
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700620 def tail(filename, n=10):
621 'Return the last n lines of a file'
622 with open(filename) as f:
623 return deque(f, n)
Raymond Hettingerd2ee64d2009-03-31 22:52:48 +0000624
625Another approach to using deques is to maintain a sequence of recently
626added elements by appending to the right and popping to the left::
627
628 def moving_average(iterable, n=3):
629 # moving_average([40, 30, 50, 46, 39, 44]) --> 40.0 42.0 45.0 43.0
630 # http://en.wikipedia.org/wiki/Moving_average
631 it = iter(iterable)
Raymond Hettingerd40285a2009-05-22 01:11:26 +0000632 d = deque(itertools.islice(it, n-1))
633 d.appendleft(0)
Raymond Hettingerd2ee64d2009-03-31 22:52:48 +0000634 s = sum(d)
Raymond Hettingerd2ee64d2009-03-31 22:52:48 +0000635 for elem in it:
636 s += elem - d.popleft()
637 d.append(elem)
638 yield s / n
639
Raymond Hettinger08584952017-11-23 13:32:23 -0800640A `round-robin scheduler
641<https://en.wikipedia.org/wiki/Round-robin_scheduling>`_ can be implemented with
642input iterators stored in a :class:`deque`. Values are yielded from the active
643iterator in position zero. If that iterator is exhausted, it can be removed
644with :meth:`~deque.popleft`; otherwise, it can be cycled back to the end with
645the :meth:`~deque.rotate` method::
646
647 def roundrobin(*iterables):
648 "roundrobin('ABC', 'D', 'EF') --> A D E B F C"
649 iterators = deque(map(iter, iterables))
650 while iterators:
651 try:
652 while True:
653 yield next(iterators[0])
654 iterators.rotate(-1)
655 except StopIteration:
656 # Remove an exhausted iterator.
657 iterators.popleft()
658
Michael Seiferte1052942018-03-26 13:40:35 +0200659The :meth:`~deque.rotate` method provides a way to implement :class:`deque` slicing and
Ezio Melotti0639d5a2009-12-19 23:26:38 +0000660deletion. For example, a pure Python implementation of ``del d[n]`` relies on
Michael Seiferte1052942018-03-26 13:40:35 +0200661the ``rotate()`` method to position elements to be popped::
Georg Brandl116aa622007-08-15 14:28:22 +0000662
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700663 def delete_nth(d, n):
664 d.rotate(-n)
665 d.popleft()
666 d.rotate(n)
Georg Brandl116aa622007-08-15 14:28:22 +0000667
668To implement :class:`deque` slicing, use a similar approach applying
Michael Seiferte1052942018-03-26 13:40:35 +0200669:meth:`~deque.rotate` to bring a target element to the left side of the deque. Remove
670old entries with :meth:`~deque.popleft`, add new entries with :meth:`~deque.extend`, and then
Georg Brandl116aa622007-08-15 14:28:22 +0000671reverse the rotation.
Georg Brandl116aa622007-08-15 14:28:22 +0000672With minor variations on that approach, it is easy to implement Forth style
673stack manipulations such as ``dup``, ``drop``, ``swap``, ``over``, ``pick``,
674``rot``, and ``roll``.
675
Georg Brandl116aa622007-08-15 14:28:22 +0000676
677:class:`defaultdict` objects
678----------------------------
679
Georg Brandl116aa622007-08-15 14:28:22 +0000680.. class:: defaultdict([default_factory[, ...]])
681
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700682 Returns a new dictionary-like object. :class:`defaultdict` is a subclass of the
683 built-in :class:`dict` class. It overrides one method and adds one writable
684 instance variable. The remaining functionality is the same as for the
685 :class:`dict` class and is not documented here.
Georg Brandl116aa622007-08-15 14:28:22 +0000686
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700687 The first argument provides the initial value for the :attr:`default_factory`
688 attribute; it defaults to ``None``. All remaining arguments are treated the same
689 as if they were passed to the :class:`dict` constructor, including keyword
690 arguments.
Georg Brandl116aa622007-08-15 14:28:22 +0000691
Georg Brandl116aa622007-08-15 14:28:22 +0000692
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700693 :class:`defaultdict` objects support the following method in addition to the
694 standard :class:`dict` operations:
Georg Brandl116aa622007-08-15 14:28:22 +0000695
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700696 .. method:: __missing__(key)
Georg Brandl116aa622007-08-15 14:28:22 +0000697
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700698 If the :attr:`default_factory` attribute is ``None``, this raises a
699 :exc:`KeyError` exception with the *key* as argument.
Georg Brandl116aa622007-08-15 14:28:22 +0000700
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700701 If :attr:`default_factory` is not ``None``, it is called without arguments
702 to provide a default value for the given *key*, this value is inserted in
703 the dictionary for the *key*, and returned.
Georg Brandl116aa622007-08-15 14:28:22 +0000704
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700705 If calling :attr:`default_factory` raises an exception this exception is
706 propagated unchanged.
Georg Brandl116aa622007-08-15 14:28:22 +0000707
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700708 This method is called by the :meth:`__getitem__` method of the
709 :class:`dict` class when the requested key is not found; whatever it
710 returns or raises is then returned or raised by :meth:`__getitem__`.
Georg Brandl116aa622007-08-15 14:28:22 +0000711
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700712 Note that :meth:`__missing__` is *not* called for any operations besides
713 :meth:`__getitem__`. This means that :meth:`get` will, like normal
714 dictionaries, return ``None`` as a default rather than using
715 :attr:`default_factory`.
Benjamin Peterson871b9d12012-01-27 09:14:01 -0500716
Georg Brandl116aa622007-08-15 14:28:22 +0000717
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700718 :class:`defaultdict` objects support the following instance variable:
Georg Brandl116aa622007-08-15 14:28:22 +0000719
Benjamin Petersone41251e2008-04-25 01:59:09 +0000720
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700721 .. attribute:: default_factory
Benjamin Petersone41251e2008-04-25 01:59:09 +0000722
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700723 This attribute is used by the :meth:`__missing__` method; it is
724 initialized from the first argument to the constructor, if present, or to
725 ``None``, if absent.
Georg Brandl116aa622007-08-15 14:28:22 +0000726
727
Georg Brandl116aa622007-08-15 14:28:22 +0000728:class:`defaultdict` Examples
729^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
730
Michael Seiferte1052942018-03-26 13:40:35 +0200731Using :class:`list` as the :attr:`~defaultdict.default_factory`, it is easy to group a
Christian Heimesfe337bf2008-03-23 21:54:12 +0000732sequence of key-value pairs into a dictionary of lists:
Georg Brandl116aa622007-08-15 14:28:22 +0000733
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700734 >>> s = [('yellow', 1), ('blue', 2), ('yellow', 3), ('blue', 4), ('red', 1)]
735 >>> d = defaultdict(list)
736 >>> for k, v in s:
737 ... d[k].append(v)
738 ...
Zachary Ware2b52c0a2016-08-09 17:38:22 -0500739 >>> sorted(d.items())
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700740 [('blue', [2, 4]), ('red', [1]), ('yellow', [1, 3])]
Georg Brandl116aa622007-08-15 14:28:22 +0000741
742When each key is encountered for the first time, it is not already in the
Michael Seiferte1052942018-03-26 13:40:35 +0200743mapping; so an entry is automatically created using the :attr:`~defaultdict.default_factory`
Georg Brandl116aa622007-08-15 14:28:22 +0000744function which returns an empty :class:`list`. The :meth:`list.append`
745operation then attaches the value to the new list. When keys are encountered
746again, the look-up proceeds normally (returning the list for that key) and the
747:meth:`list.append` operation adds another value to the list. This technique is
Christian Heimesfe337bf2008-03-23 21:54:12 +0000748simpler and faster than an equivalent technique using :meth:`dict.setdefault`:
Georg Brandl116aa622007-08-15 14:28:22 +0000749
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700750 >>> d = {}
751 >>> for k, v in s:
752 ... d.setdefault(k, []).append(v)
753 ...
Zachary Ware2b52c0a2016-08-09 17:38:22 -0500754 >>> sorted(d.items())
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700755 [('blue', [2, 4]), ('red', [1]), ('yellow', [1, 3])]
Georg Brandl116aa622007-08-15 14:28:22 +0000756
Michael Seiferte1052942018-03-26 13:40:35 +0200757Setting the :attr:`~defaultdict.default_factory` to :class:`int` makes the
Georg Brandl116aa622007-08-15 14:28:22 +0000758:class:`defaultdict` useful for counting (like a bag or multiset in other
Christian Heimesfe337bf2008-03-23 21:54:12 +0000759languages):
Georg Brandl116aa622007-08-15 14:28:22 +0000760
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700761 >>> s = 'mississippi'
762 >>> d = defaultdict(int)
763 >>> for k in s:
764 ... d[k] += 1
765 ...
Zachary Ware2b52c0a2016-08-09 17:38:22 -0500766 >>> sorted(d.items())
767 [('i', 4), ('m', 1), ('p', 2), ('s', 4)]
Georg Brandl116aa622007-08-15 14:28:22 +0000768
769When a letter is first encountered, it is missing from the mapping, so the
Michael Seiferte1052942018-03-26 13:40:35 +0200770:attr:`~defaultdict.default_factory` function calls :func:`int` to supply a default count of
Georg Brandl116aa622007-08-15 14:28:22 +0000771zero. The increment operation then builds up the count for each letter.
772
773The function :func:`int` which always returns zero is just a special case of
774constant functions. A faster and more flexible way to create constant functions
775is to use a lambda function which can supply any constant value (not just
Christian Heimesfe337bf2008-03-23 21:54:12 +0000776zero):
Georg Brandl116aa622007-08-15 14:28:22 +0000777
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700778 >>> def constant_factory(value):
779 ... return lambda: value
780 >>> d = defaultdict(constant_factory('<missing>'))
781 >>> d.update(name='John', action='ran')
782 >>> '%(name)s %(action)s to %(object)s' % d
783 'John ran to <missing>'
Georg Brandl116aa622007-08-15 14:28:22 +0000784
Michael Seiferte1052942018-03-26 13:40:35 +0200785Setting the :attr:`~defaultdict.default_factory` to :class:`set` makes the
Christian Heimesfe337bf2008-03-23 21:54:12 +0000786:class:`defaultdict` useful for building a dictionary of sets:
Georg Brandl116aa622007-08-15 14:28:22 +0000787
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700788 >>> s = [('red', 1), ('blue', 2), ('red', 3), ('blue', 4), ('red', 1), ('blue', 4)]
789 >>> d = defaultdict(set)
790 >>> for k, v in s:
791 ... d[k].add(v)
792 ...
Zachary Ware2b52c0a2016-08-09 17:38:22 -0500793 >>> sorted(d.items())
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700794 [('blue', {2, 4}), ('red', {1, 3})]
Georg Brandl116aa622007-08-15 14:28:22 +0000795
796
Guido van Rossum3d392eb2007-11-16 00:35:22 +0000797:func:`namedtuple` Factory Function for Tuples with Named Fields
Christian Heimes790c8232008-01-07 21:14:23 +0000798----------------------------------------------------------------
Georg Brandl116aa622007-08-15 14:28:22 +0000799
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000800Named tuples assign meaning to each position in a tuple and allow for more readable,
801self-documenting code. They can be used wherever regular tuples are used, and
802they add the ability to access fields by name instead of position index.
Georg Brandl116aa622007-08-15 14:28:22 +0000803
Raymond Hettinger39482072018-01-10 21:45:19 -0800804.. function:: namedtuple(typename, field_names, *, rename=False, defaults=None, module=None)
Georg Brandl116aa622007-08-15 14:28:22 +0000805
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700806 Returns a new tuple subclass named *typename*. The new subclass is used to
807 create tuple-like objects that have fields accessible by attribute lookup as
808 well as being indexable and iterable. Instances of the subclass also have a
809 helpful docstring (with typename and field_names) and a helpful :meth:`__repr__`
810 method which lists the tuple contents in a ``name=value`` format.
Georg Brandl116aa622007-08-15 14:28:22 +0000811
csabella97bf7222017-04-25 12:14:45 -0400812 The *field_names* are a sequence of strings such as ``['x', 'y']``.
813 Alternatively, *field_names* can be a single string with each fieldname
814 separated by whitespace and/or commas, for example ``'x y'`` or ``'x, y'``.
Georg Brandl9afde1c2007-11-01 20:32:30 +0000815
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700816 Any valid Python identifier may be used for a fieldname except for names
817 starting with an underscore. Valid identifiers consist of letters, digits,
818 and underscores but do not start with a digit or underscore and cannot be
819 a :mod:`keyword` such as *class*, *for*, *return*, *global*, *pass*,
820 or *raise*.
Georg Brandl116aa622007-08-15 14:28:22 +0000821
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700822 If *rename* is true, invalid fieldnames are automatically replaced
823 with positional names. For example, ``['abc', 'def', 'ghi', 'abc']`` is
824 converted to ``['abc', '_1', 'ghi', '_3']``, eliminating the keyword
825 ``def`` and the duplicate fieldname ``abc``.
Benjamin Petersona86f2c02009-02-10 02:41:10 +0000826
Raymond Hettinger39482072018-01-10 21:45:19 -0800827 *defaults* can be ``None`` or an :term:`iterable` of default values.
828 Since fields with a default value must come after any fields without a
829 default, the *defaults* are applied to the rightmost parameters. For
830 example, if the fieldnames are ``['x', 'y', 'z']`` and the defaults are
831 ``(1, 2)``, then ``x`` will be a required argument, ``y`` will default to
832 ``1``, and ``z`` will default to ``2``.
833
Raymond Hettinger0d5048c2016-09-12 00:18:31 -0700834 If *module* is defined, the ``__module__`` attribute of the named tuple is
835 set to that value.
836
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700837 Named tuple instances do not have per-instance dictionaries, so they are
838 lightweight and require no more memory than regular tuples.
Georg Brandl116aa622007-08-15 14:28:22 +0000839
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700840 .. versionchanged:: 3.1
Raymond Hettinger6538b432016-08-16 10:55:43 -0700841 Added support for *rename*.
842
843 .. versionchanged:: 3.6
844 The *verbose* and *rename* parameters became
845 :ref:`keyword-only arguments <keyword-only_parameter>`.
Benjamin Petersona86f2c02009-02-10 02:41:10 +0000846
Raymond Hettinger0d5048c2016-09-12 00:18:31 -0700847 .. versionchanged:: 3.6
848 Added the *module* parameter.
Christian Heimesfe337bf2008-03-23 21:54:12 +0000849
Raymond Hettinger8b57d732017-09-10 10:23:36 -0700850 .. versionchanged:: 3.7
851 Remove the *verbose* parameter and the :attr:`_source` attribute.
852
Raymond Hettinger39482072018-01-10 21:45:19 -0800853 .. versionchanged:: 3.7
854 Added the *defaults* parameter and the :attr:`_field_defaults`
855 attribute.
856
Christian Heimesfe337bf2008-03-23 21:54:12 +0000857.. doctest::
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700858 :options: +NORMALIZE_WHITESPACE
Georg Brandl116aa622007-08-15 14:28:22 +0000859
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700860 >>> # Basic example
861 >>> Point = namedtuple('Point', ['x', 'y'])
862 >>> p = Point(11, y=22) # instantiate with positional or keyword arguments
863 >>> p[0] + p[1] # indexable like the plain tuple (11, 22)
864 33
865 >>> x, y = p # unpack like a regular tuple
866 >>> x, y
867 (11, 22)
868 >>> p.x + p.y # fields also accessible by name
869 33
870 >>> p # readable __repr__ with a name=value style
871 Point(x=11, y=22)
Georg Brandl116aa622007-08-15 14:28:22 +0000872
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000873Named tuples are especially useful for assigning field names to result tuples returned
874by the :mod:`csv` or :mod:`sqlite3` modules::
875
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700876 EmployeeRecord = namedtuple('EmployeeRecord', 'name, age, title, department, paygrade')
Georg Brandl9afde1c2007-11-01 20:32:30 +0000877
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700878 import csv
879 for emp in map(EmployeeRecord._make, csv.reader(open("employees.csv", "rb"))):
880 print(emp.name, emp.title)
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000881
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700882 import sqlite3
883 conn = sqlite3.connect('/companydata')
884 cursor = conn.cursor()
885 cursor.execute('SELECT name, age, title, department, paygrade FROM employees')
886 for emp in map(EmployeeRecord._make, cursor.fetchall()):
887 print(emp.name, emp.title)
Georg Brandl9afde1c2007-11-01 20:32:30 +0000888
Christian Heimes99170a52007-12-19 02:07:34 +0000889In addition to the methods inherited from tuples, named tuples support
Raymond Hettinger2ebea412011-03-23 12:52:23 -0700890three additional methods and two attributes. To prevent conflicts with
Christian Heimes2380ac72008-01-09 00:17:24 +0000891field names, the method and attribute names start with an underscore.
Christian Heimes99170a52007-12-19 02:07:34 +0000892
Benjamin Peterson0b9fb802010-07-18 14:23:36 +0000893.. classmethod:: somenamedtuple._make(iterable)
Christian Heimes99170a52007-12-19 02:07:34 +0000894
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700895 Class method that makes a new instance from an existing sequence or iterable.
Christian Heimes99170a52007-12-19 02:07:34 +0000896
Raymond Hettinger6fed9fd2012-06-11 00:38:14 -0700897 .. doctest::
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000898
Raymond Hettinger6fed9fd2012-06-11 00:38:14 -0700899 >>> t = [11, 22]
900 >>> Point._make(t)
901 Point(x=11, y=22)
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000902
Christian Heimes790c8232008-01-07 21:14:23 +0000903.. method:: somenamedtuple._asdict()
Georg Brandl9afde1c2007-11-01 20:32:30 +0000904
Raymond Hettinger49fd6dd2019-02-21 00:05:30 -0800905 Return a new :class:`dict` which maps field names to their corresponding
Raymond Hettingerfd27f622016-08-16 13:13:17 -0700906 values:
907
908 .. doctest::
Georg Brandl9afde1c2007-11-01 20:32:30 +0000909
Raymond Hettinger7a3602e2015-08-30 09:13:48 -0700910 >>> p = Point(x=11, y=22)
911 >>> p._asdict()
Raymond Hettinger0bb4bdf2019-01-31 00:59:50 -0800912 {'x': 11, 'y': 22}
Raymond Hettingera4f52b12009-03-02 22:28:31 +0000913
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700914 .. versionchanged:: 3.1
915 Returns an :class:`OrderedDict` instead of a regular :class:`dict`.
Christian Heimesfe337bf2008-03-23 21:54:12 +0000916
Raymond Hettinger0bb4bdf2019-01-31 00:59:50 -0800917 .. versionchanged:: 3.8
918 Returns a regular :class:`dict` instead of an :class:`OrderedDict`.
919 As of Python 3.7, regular dicts are guaranteed to be ordered. If the
920 extra features of :class:`OrderedDict` are required, the suggested
921 remediation is to cast the result to the desired type:
922 ``OrderedDict(nt._asdict())``.
923
Ben Hoyt184bd822017-06-13 15:20:51 -0400924.. method:: somenamedtuple._replace(**kwargs)
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000925
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700926 Return a new instance of the named tuple replacing specified fields with new
Raymond Hettinger6fed9fd2012-06-11 00:38:14 -0700927 values::
Thomas Wouters8ce81f72007-09-20 18:22:40 +0000928
Raymond Hettinger6fed9fd2012-06-11 00:38:14 -0700929 >>> p = Point(x=11, y=22)
930 >>> p._replace(x=33)
931 Point(x=33, y=22)
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000932
Raymond Hettinger6fed9fd2012-06-11 00:38:14 -0700933 >>> for partnum, record in inventory.items():
934 ... inventory[partnum] = record._replace(price=newprices[partnum], timestamp=time.now())
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000935
Christian Heimes790c8232008-01-07 21:14:23 +0000936.. attribute:: somenamedtuple._fields
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000937
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700938 Tuple of strings listing the field names. Useful for introspection
939 and for creating new named tuple types from existing named tuples.
Thomas Wouters8ce81f72007-09-20 18:22:40 +0000940
Raymond Hettinger6fed9fd2012-06-11 00:38:14 -0700941 .. doctest::
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000942
Raymond Hettinger6fed9fd2012-06-11 00:38:14 -0700943 >>> p._fields # view the field names
944 ('x', 'y')
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000945
Raymond Hettinger6fed9fd2012-06-11 00:38:14 -0700946 >>> Color = namedtuple('Color', 'red green blue')
947 >>> Pixel = namedtuple('Pixel', Point._fields + Color._fields)
948 >>> Pixel(11, 22, 128, 255, 0)
949 Pixel(x=11, y=22, red=128, green=255, blue=0)
Georg Brandl116aa622007-08-15 14:28:22 +0000950
Raymond Hettinger39482072018-01-10 21:45:19 -0800951.. attribute:: somenamedtuple._fields_defaults
952
953 Dictionary mapping field names to default values.
954
955 .. doctest::
956
957 >>> Account = namedtuple('Account', ['type', 'balance'], defaults=[0])
958 >>> Account._fields_defaults
959 {'balance': 0}
960 >>> Account('premium')
961 Account(type='premium', balance=0)
962
Christian Heimes0449f632007-12-15 01:27:15 +0000963To retrieve a field whose name is stored in a string, use the :func:`getattr`
Christian Heimesfe337bf2008-03-23 21:54:12 +0000964function:
Christian Heimes0449f632007-12-15 01:27:15 +0000965
966 >>> getattr(p, 'x')
967 11
968
Raymond Hettinger651453a2009-02-11 00:20:02 +0000969To convert a dictionary to a named tuple, use the double-star-operator
970(as described in :ref:`tut-unpacking-arguments`):
Christian Heimes99170a52007-12-19 02:07:34 +0000971
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700972 >>> d = {'x': 11, 'y': 22}
973 >>> Point(**d)
974 Point(x=11, y=22)
Christian Heimes99170a52007-12-19 02:07:34 +0000975
Guido van Rossum3d392eb2007-11-16 00:35:22 +0000976Since a named tuple is a regular Python class, it is easy to add or change
Christian Heimes043d6f62008-01-07 17:19:16 +0000977functionality with a subclass. Here is how to add a calculated field and
Christian Heimesfe337bf2008-03-23 21:54:12 +0000978a fixed-width print format:
Guido van Rossum3d392eb2007-11-16 00:35:22 +0000979
Raymond Hettingerfd27f622016-08-16 13:13:17 -0700980.. doctest::
981
982 >>> class Point(namedtuple('Point', ['x', 'y'])):
Zachary Ware2b52c0a2016-08-09 17:38:22 -0500983 ... __slots__ = ()
984 ... @property
985 ... def hypot(self):
986 ... return (self.x ** 2 + self.y ** 2) ** 0.5
987 ... def __str__(self):
988 ... return 'Point: x=%6.3f y=%6.3f hypot=%6.3f' % (self.x, self.y, self.hypot)
Guido van Rossum3d392eb2007-11-16 00:35:22 +0000989
Georg Brandl0df79792008-10-04 18:33:26 +0000990 >>> for p in Point(3, 4), Point(14, 5/7):
Zachary Ware2b52c0a2016-08-09 17:38:22 -0500991 ... print(p)
Christian Heimes25bb7832008-01-11 16:17:00 +0000992 Point: x= 3.000 y= 4.000 hypot= 5.000
993 Point: x=14.000 y= 0.714 hypot=14.018
Christian Heimes043d6f62008-01-07 17:19:16 +0000994
Georg Brandlaf5c2382009-12-28 08:02:38 +0000995The subclass shown above sets ``__slots__`` to an empty tuple. This helps
Christian Heimes679db4a2008-01-18 09:56:22 +0000996keep memory requirements low by preventing the creation of instance dictionaries.
997
Christian Heimes2380ac72008-01-09 00:17:24 +0000998Subclassing is not useful for adding new, stored fields. Instead, simply
Michael Seiferte1052942018-03-26 13:40:35 +0200999create a new named tuple type from the :attr:`~somenamedtuple._fields` attribute:
Christian Heimes2380ac72008-01-09 00:17:24 +00001000
Christian Heimes25bb7832008-01-11 16:17:00 +00001001 >>> Point3D = namedtuple('Point3D', Point._fields + ('z',))
Christian Heimes2380ac72008-01-09 00:17:24 +00001002
Raymond Hettingereac503a2015-05-13 01:09:59 -07001003Docstrings can be customized by making direct assignments to the ``__doc__``
1004fields:
1005
1006 >>> Book = namedtuple('Book', ['id', 'title', 'authors'])
Raymond Hettinger850be0f2015-11-09 08:24:53 -08001007 >>> Book.__doc__ += ': Hardcover book in active collection'
Berker Peksagde7cafa2015-05-13 12:16:27 +03001008 >>> Book.id.__doc__ = '13-digit ISBN'
1009 >>> Book.title.__doc__ = 'Title of first printing'
Raymond Hettinger850be0f2015-11-09 08:24:53 -08001010 >>> Book.authors.__doc__ = 'List of authors sorted by last name'
Raymond Hettingereac503a2015-05-13 01:09:59 -07001011
Raymond Hettinger6e701312015-11-23 22:18:55 -08001012.. versionchanged:: 3.5
1013 Property docstrings became writeable.
1014
Michael Seiferte1052942018-03-26 13:40:35 +02001015Default values can be implemented by using :meth:`~somenamedtuple._replace` to
Christian Heimesfe337bf2008-03-23 21:54:12 +00001016customize a prototype instance:
Guido van Rossum3d392eb2007-11-16 00:35:22 +00001017
1018 >>> Account = namedtuple('Account', 'owner balance transaction_count')
Christian Heimes587c2bf2008-01-19 16:21:02 +00001019 >>> default_account = Account('<owner name>', 0.0, 0)
1020 >>> johns_account = default_account._replace(owner='John')
Raymond Hettingerb2d09452011-03-22 22:36:21 -07001021 >>> janes_account = default_account._replace(owner='Jane')
Guido van Rossum3d392eb2007-11-16 00:35:22 +00001022
Georg Brandl8ed75cd2014-10-31 10:25:48 +01001023
1024.. seealso::
1025
Raymond Hettinger7929cfb2012-06-09 19:15:26 -07001026 * `Recipe for named tuple abstract base class with a metaclass mix-in
Serhiy Storchaka6dff0202016-05-07 10:49:07 +03001027 <https://code.activestate.com/recipes/577629-namedtupleabc-abstract-base-class-mix-in-for-named/>`_
Raymond Hettingerbfcb4292012-06-10 11:39:44 -07001028 by Jan Kaliszewski. Besides providing an :term:`abstract base class` for
1029 named tuples, it also supports an alternate :term:`metaclass`-based
1030 constructor that is convenient for use cases where named tuples are being
1031 subclassed.
Raymond Hettingere4c96ad2008-02-06 01:23:58 +00001032
Raymond Hettingerfd27f622016-08-16 13:13:17 -07001033 * See :meth:`types.SimpleNamespace` for a mutable namespace based on an
1034 underlying dictionary instead of a tuple.
1035
1036 * See :meth:`typing.NamedTuple` for a way to add type hints for named tuples.
Raymond Hettinger2a75e8f2015-08-16 08:32:01 -07001037
Raymond Hettingere4c96ad2008-02-06 01:23:58 +00001038
Raymond Hettinger2d32f632009-03-02 21:24:57 +00001039:class:`OrderedDict` objects
1040----------------------------
1041
Raymond Hettinger49fd6dd2019-02-21 00:05:30 -08001042Ordered dictionaries are just like regular dictionaries but have some extra
1043capabilities relating to ordering operations. They have become less
1044important now that the built-in :class:`dict` class gained the ability
1045to remember insertion order (this new behavior became guaranteed in
1046Python 3.7).
1047
1048Some differences from :class:`dict` still remain:
1049
1050* The regular :class:`dict` was designed to be very good at mapping
1051 operations. Tracking insertion order was secondary.
1052
1053* The :class:`OrderedDict` was designed to be good at reordering operations.
1054 Space efficiency, iteration speed, and the performance of update
1055 operations were secondary.
1056
1057* Algorithmically, :class:`OrderedDict` can handle frequent reordering
1058 operations better than :class:`dict`. This makes it suitable for tracking
1059 recent accesses (for example in an `LRU cache
1060 <https://medium.com/@krishankantsinghal/my-first-blog-on-medium-583159139237>`_).
1061
1062* The equality operation for :class:`OrderedDict` checks for matching order.
1063
1064* The :meth:`popitem` method of :class:`OrderedDict` has a different
1065 signature. It accepts an optional argument to specify which item is popped.
1066
1067* :class:`OrderedDict` has a :meth:`move_to_end` method to
1068 efficiently reposition an element to an endpoint.
1069
1070* Until Python 3.8, :class:`dict` lacked a :meth:`__reversed__` method.
1071
Raymond Hettinger2d32f632009-03-02 21:24:57 +00001072
1073.. class:: OrderedDict([items])
1074
Raymond Hettinger49fd6dd2019-02-21 00:05:30 -08001075 Return an instance of a :class:`dict` subclass that has methods
1076 specialized for rearranging dictionary order.
Raymond Hettinger2d32f632009-03-02 21:24:57 +00001077
Raymond Hettinger7929cfb2012-06-09 19:15:26 -07001078 .. versionadded:: 3.1
Raymond Hettinger2d32f632009-03-02 21:24:57 +00001079
Raymond Hettinger7929cfb2012-06-09 19:15:26 -07001080 .. method:: popitem(last=True)
Raymond Hettingerdc879f02009-03-19 20:30:56 +00001081
Raymond Hettinger7929cfb2012-06-09 19:15:26 -07001082 The :meth:`popitem` method for ordered dictionaries returns and removes a
Serhiy Storchaka4ecfa452016-05-16 09:31:54 +03001083 (key, value) pair. The pairs are returned in
1084 :abbr:`LIFO (last-in, first-out)` order if *last* is true
1085 or :abbr:`FIFO (first-in, first-out)` order if false.
Raymond Hettinger2d32f632009-03-02 21:24:57 +00001086
Raymond Hettinger7929cfb2012-06-09 19:15:26 -07001087 .. method:: move_to_end(key, last=True)
Raymond Hettingerf45abc92010-09-06 21:26:09 +00001088
Raymond Hettinger7929cfb2012-06-09 19:15:26 -07001089 Move an existing *key* to either end of an ordered dictionary. The item
1090 is moved to the right end if *last* is true (the default) or to the
1091 beginning if *last* is false. Raises :exc:`KeyError` if the *key* does
1092 not exist::
Raymond Hettingerf45abc92010-09-06 21:26:09 +00001093
Raymond Hettinger7929cfb2012-06-09 19:15:26 -07001094 >>> d = OrderedDict.fromkeys('abcde')
1095 >>> d.move_to_end('b')
1096 >>> ''.join(d.keys())
1097 'acdeb'
1098 >>> d.move_to_end('b', last=False)
1099 >>> ''.join(d.keys())
1100 'bacde'
Raymond Hettingerf45abc92010-09-06 21:26:09 +00001101
Raymond Hettinger7929cfb2012-06-09 19:15:26 -07001102 .. versionadded:: 3.2
Raymond Hettingerf45abc92010-09-06 21:26:09 +00001103
Raymond Hettingere9091502009-05-19 17:40:07 +00001104In addition to the usual mapping methods, ordered dictionaries also support
1105reverse iteration using :func:`reversed`.
1106
Raymond Hettinger2d32f632009-03-02 21:24:57 +00001107Equality tests between :class:`OrderedDict` objects are order-sensitive
1108and are implemented as ``list(od1.items())==list(od2.items())``.
1109Equality tests between :class:`OrderedDict` objects and other
Serhiy Storchakabfdcd432013-10-13 23:09:14 +03001110:class:`~collections.abc.Mapping` objects are order-insensitive like regular
1111dictionaries. This allows :class:`OrderedDict` objects to be substituted
1112anywhere a regular dictionary is used.
Raymond Hettinger2d32f632009-03-02 21:24:57 +00001113
Serhiy Storchaka578c9212014-04-04 15:19:36 +03001114.. versionchanged:: 3.5
Martin Panter397625e2015-10-07 10:03:20 +00001115 The items, keys, and values :term:`views <dictionary view>`
1116 of :class:`OrderedDict` now support reverse iteration using :func:`reversed`.
Raymond Hettingerdc879f02009-03-19 20:30:56 +00001117
Raymond Hettingerd15bb262017-01-07 22:05:12 -08001118.. versionchanged:: 3.6
1119 With the acceptance of :pep:`468`, order is retained for keyword arguments
1120 passed to the :class:`OrderedDict` constructor and its :meth:`update`
1121 method.
1122
Raymond Hettinger7bba6832011-04-15 17:43:19 -07001123:class:`OrderedDict` Examples and Recipes
1124^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
1125
Raymond Hettinger49fd6dd2019-02-21 00:05:30 -08001126It is straightforward to create an ordered dictionary variant
Andrew Svetlovff63e7a2012-08-31 13:54:54 +03001127that remembers the order the keys were *last* inserted.
Raymond Hettinger4821ef82010-07-31 10:14:41 +00001128If a new entry overwrites an existing entry, the
1129original insertion position is changed and moved to the end::
1130
1131 class LastUpdatedOrderedDict(OrderedDict):
Georg Brandl77570e22010-12-18 16:21:58 +00001132 'Store items in the order the keys were last added'
Raymond Hettinger7bba6832011-04-15 17:43:19 -07001133
Raymond Hettinger4821ef82010-07-31 10:14:41 +00001134 def __setitem__(self, key, value):
Raymond Hettinger49fd6dd2019-02-21 00:05:30 -08001135 super().__setitem__(key, value)
1136 super().move_to_end(key)
Raymond Hettinger4821ef82010-07-31 10:14:41 +00001137
Raymond Hettinger49fd6dd2019-02-21 00:05:30 -08001138An :class:`OrderedDict` would also be useful for implementing
1139variants of :func:`functools.lru_cache`::
Raymond Hettinger7bba6832011-04-15 17:43:19 -07001140
Raymond Hettinger49fd6dd2019-02-21 00:05:30 -08001141 class LRU(OrderedDict):
1142 'Limit size, evicting the least recently looked-up key when full'
Raymond Hettinger7bba6832011-04-15 17:43:19 -07001143
Raymond Hettinger49fd6dd2019-02-21 00:05:30 -08001144 def __init__(self, maxsize=128, *args, **kwds):
1145 self.maxsize = maxsize
1146 super().__init__(*args, **kwds)
Raymond Hettinger7bba6832011-04-15 17:43:19 -07001147
Raymond Hettinger49fd6dd2019-02-21 00:05:30 -08001148 def __getitem__(self, key):
1149 value = super().__getitem__(key)
1150 self.move_to_end(key)
1151 return value
1152
1153 def __setitem__(self, key, value):
1154 super().__setitem__(key, value)
1155 if len(self) > self.maxsize:
1156 oldest = next(iter(self))
1157 del self[oldest]
Raymond Hettinger7bba6832011-04-15 17:43:19 -07001158
Raymond Hettingere4c96ad2008-02-06 01:23:58 +00001159
1160:class:`UserDict` objects
Mark Summerfield8f2d0062008-02-06 13:30:44 +00001161-------------------------
Raymond Hettingere4c96ad2008-02-06 01:23:58 +00001162
Georg Brandl48310cd2009-01-03 21:18:54 +00001163The class, :class:`UserDict` acts as a wrapper around dictionary objects.
1164The need for this class has been partially supplanted by the ability to
Raymond Hettingere4c96ad2008-02-06 01:23:58 +00001165subclass directly from :class:`dict`; however, this class can be easier
1166to work with because the underlying dictionary is accessible as an
1167attribute.
1168
1169.. class:: UserDict([initialdata])
1170
Raymond Hettinger7929cfb2012-06-09 19:15:26 -07001171 Class that simulates a dictionary. The instance's contents are kept in a
1172 regular dictionary, which is accessible via the :attr:`data` attribute of
1173 :class:`UserDict` instances. If *initialdata* is provided, :attr:`data` is
1174 initialized with its contents; note that a reference to *initialdata* will not
1175 be kept, allowing it be used for other purposes.
Raymond Hettingere4c96ad2008-02-06 01:23:58 +00001176
Raymond Hettinger7929cfb2012-06-09 19:15:26 -07001177 In addition to supporting the methods and operations of mappings,
1178 :class:`UserDict` instances provide the following attribute:
Raymond Hettingere4c96ad2008-02-06 01:23:58 +00001179
Raymond Hettinger7929cfb2012-06-09 19:15:26 -07001180 .. attribute:: data
Raymond Hettingere4c96ad2008-02-06 01:23:58 +00001181
Raymond Hettinger7929cfb2012-06-09 19:15:26 -07001182 A real dictionary used to store the contents of the :class:`UserDict`
1183 class.
Raymond Hettinger53dbe392008-02-12 20:03:09 +00001184
1185
1186
1187:class:`UserList` objects
1188-------------------------
1189
1190This class acts as a wrapper around list objects. It is a useful base class
Georg Brandl48310cd2009-01-03 21:18:54 +00001191for your own list-like classes which can inherit from them and override
Raymond Hettinger53dbe392008-02-12 20:03:09 +00001192existing methods or add new ones. In this way, one can add new behaviors to
1193lists.
1194
Georg Brandl48310cd2009-01-03 21:18:54 +00001195The need for this class has been partially supplanted by the ability to
Raymond Hettinger53dbe392008-02-12 20:03:09 +00001196subclass directly from :class:`list`; however, this class can be easier
1197to work with because the underlying list is accessible as an attribute.
1198
1199.. class:: UserList([list])
1200
Raymond Hettinger7929cfb2012-06-09 19:15:26 -07001201 Class that simulates a list. The instance's contents are kept in a regular
1202 list, which is accessible via the :attr:`data` attribute of :class:`UserList`
1203 instances. The instance's contents are initially set to a copy of *list*,
1204 defaulting to the empty list ``[]``. *list* can be any iterable, for
1205 example a real Python list or a :class:`UserList` object.
Raymond Hettinger53dbe392008-02-12 20:03:09 +00001206
Raymond Hettinger7929cfb2012-06-09 19:15:26 -07001207 In addition to supporting the methods and operations of mutable sequences,
1208 :class:`UserList` instances provide the following attribute:
Raymond Hettinger53dbe392008-02-12 20:03:09 +00001209
Raymond Hettinger7929cfb2012-06-09 19:15:26 -07001210 .. attribute:: data
Raymond Hettinger53dbe392008-02-12 20:03:09 +00001211
Raymond Hettinger7929cfb2012-06-09 19:15:26 -07001212 A real :class:`list` object used to store the contents of the
1213 :class:`UserList` class.
Raymond Hettinger53dbe392008-02-12 20:03:09 +00001214
Zachary Ware80602e02014-01-13 20:38:57 -06001215**Subclassing requirements:** Subclasses of :class:`UserList` are expected to
Raymond Hettinger53dbe392008-02-12 20:03:09 +00001216offer a constructor which can be called with either no arguments or one
1217argument. List operations which return a new sequence attempt to create an
1218instance of the actual implementation class. To do so, it assumes that the
1219constructor can be called with a single parameter, which is a sequence object
1220used as a data source.
1221
1222If a derived class does not wish to comply with this requirement, all of the
1223special methods supported by this class will need to be overridden; please
1224consult the sources for information about the methods which need to be provided
1225in that case.
Raymond Hettingerb3a65f82008-02-21 22:11:37 +00001226
1227:class:`UserString` objects
Christian Heimesc3f30c42008-02-22 16:37:40 +00001228---------------------------
Raymond Hettingerb3a65f82008-02-21 22:11:37 +00001229
Georg Brandl48310cd2009-01-03 21:18:54 +00001230The class, :class:`UserString` acts as a wrapper around string objects.
1231The need for this class has been partially supplanted by the ability to
Raymond Hettingerb3a65f82008-02-21 22:11:37 +00001232subclass directly from :class:`str`; however, this class can be easier
1233to work with because the underlying string is accessible as an
1234attribute.
1235
Michael Seiferte1052942018-03-26 13:40:35 +02001236.. class:: UserString(seq)
Raymond Hettingerb3a65f82008-02-21 22:11:37 +00001237
Michael Seiferte1052942018-03-26 13:40:35 +02001238 Class that simulates a string object. The instance's
Raymond Hettinger7929cfb2012-06-09 19:15:26 -07001239 content is kept in a regular string object, which is accessible via the
1240 :attr:`data` attribute of :class:`UserString` instances. The instance's
Michael Seiferte1052942018-03-26 13:40:35 +02001241 contents are initially set to a copy of *seq*. The *seq* argument can
1242 be any object which can be converted into a string using the built-in
1243 :func:`str` function.
1244
1245 In addition to supporting the methods and operations of strings,
1246 :class:`UserString` instances provide the following attribute:
1247
1248 .. attribute:: data
1249
1250 A real :class:`str` object used to store the contents of the
1251 :class:`UserString` class.
Yury Selivanov336b37b2015-09-09 12:23:01 -04001252
1253 .. versionchanged:: 3.5
1254 New methods ``__getnewargs__``, ``__rmod__``, ``casefold``,
1255 ``format_map``, ``isprintable``, and ``maketrans``.