blob: f2befe7c08e25ad9a7941e538f02992342d3196c [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
Georg Brandl116aa622007-08-15 14:28:22 +00006.. moduleauthor:: Raymond Hettinger <python@rcn.com>
7.. sectionauthor:: Raymond Hettinger <python@rcn.com>
8
Christian Heimesfe337bf2008-03-23 21:54:12 +00009.. testsetup:: *
10
Raymond Hettinger7929cfb2012-06-09 19:15:26 -070011 from collections import *
12 import itertools
13 __name__ = '<doctest>'
Georg Brandl116aa622007-08-15 14:28:22 +000014
Raymond Hettinger158c9c22011-02-22 00:41:50 +000015**Source code:** :source:`Lib/collections/__init__.py`
Raymond Hettinger10480942011-01-10 03:26:08 +000016
Raymond Hettinger4f707fd2011-01-10 19:54:11 +000017--------------
18
Raymond Hettingera6b76ba2010-08-08 00:29:08 +000019This module implements specialized container datatypes providing alternatives to
20Python's general purpose built-in containers, :class:`dict`, :class:`list`,
21:class:`set`, and :class:`tuple`.
Christian Heimes0bd4e112008-02-12 22:59:25 +000022
Raymond Hettingera6b76ba2010-08-08 00:29:08 +000023===================== ====================================================================
24:func:`namedtuple` factory function for creating tuple subclasses with named fields
25:class:`deque` list-like container with fast appends and pops on either end
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +000026:class:`ChainMap` dict-like class for creating a single view of multiple mappings
Raymond Hettingera6b76ba2010-08-08 00:29:08 +000027:class:`Counter` dict subclass for counting hashable objects
28:class:`OrderedDict` dict subclass that remembers the order entries were added
29:class:`defaultdict` dict subclass that calls a factory function to supply missing values
30:class:`UserDict` wrapper around dictionary objects for easier dict subclassing
31:class:`UserList` wrapper around list objects for easier list subclassing
32:class:`UserString` wrapper around string objects for easier string subclassing
33===================== ====================================================================
Georg Brandl116aa622007-08-15 14:28:22 +000034
Raymond Hettinger158c9c22011-02-22 00:41:50 +000035.. versionchanged:: 3.3
Raymond Hettinger7929cfb2012-06-09 19:15:26 -070036 Moved :ref:`collections-abstract-base-classes` to the :mod:`collections.abc` module.
37 For backwards compatibility, they continue to be visible in this module
38 as well.
Mark Summerfield08898b42007-09-05 08:43:04 +000039
40
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +000041:class:`ChainMap` objects
42-------------------------
43
Georg Brandl283b96b2012-04-03 09:16:46 +020044.. versionadded:: 3.3
45
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +000046A :class:`ChainMap` class is provided for quickly linking a number of mappings
47so they can be treated as a single unit. It is often much faster than creating
48a new dictionary and running multiple :meth:`~dict.update` calls.
49
50The class can be used to simulate nested scopes and is useful in templating.
51
52.. class:: ChainMap(*maps)
53
Raymond Hettinger7929cfb2012-06-09 19:15:26 -070054 A :class:`ChainMap` groups multiple dicts or other mappings together to
55 create a single, updateable view. If no *maps* are specified, a single empty
56 dictionary is provided so that a new chain always has at least one mapping.
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +000057
Raymond Hettinger7929cfb2012-06-09 19:15:26 -070058 The underlying mappings are stored in a list. That list is public and can
59 accessed or updated using the *maps* attribute. There is no other state.
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +000060
Raymond Hettinger7929cfb2012-06-09 19:15:26 -070061 Lookups search the underlying mappings successively until a key is found. In
62 contrast, writes, updates, and deletions only operate on the first mapping.
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +000063
Raymond Hettinger7929cfb2012-06-09 19:15:26 -070064 A :class:`ChainMap` incorporates the underlying mappings by reference. So, if
65 one of the underlying mappings gets updated, those changes will be reflected
66 in :class:`ChainMap`.
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +000067
Raymond Hettinger7929cfb2012-06-09 19:15:26 -070068 All of the usual dictionary methods are supported. In addition, there is a
69 *maps* attribute, a method for creating new subcontexts, and a property for
70 accessing all but the first mapping:
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +000071
Raymond Hettinger7929cfb2012-06-09 19:15:26 -070072 .. attribute:: maps
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +000073
Raymond Hettinger7929cfb2012-06-09 19:15:26 -070074 A user updateable list of mappings. The list is ordered from
75 first-searched to last-searched. It is the only stored state and can
76 be modified to change which mappings are searched. The list should
77 always contain at least one mapping.
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +000078
Vinay Sajip1ba81ee2013-01-11 23:39:53 +000079 .. method:: new_child(m=None)
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +000080
Vinay Sajip1ba81ee2013-01-11 23:39:53 +000081 Returns a new :class:`ChainMap` containing a new map followed by
82 all of the maps in the current instance. If ``m`` is specified,
83 it becomes the new map at the front of the list of mappings; if not
84 specified, an empty dict is used, so that a call to ``d.new_child()``
85 is equivalent to: ``ChainMap({}, *d.maps)``. This method is used for
Raymond Hettinger7929cfb2012-06-09 19:15:26 -070086 creating subcontexts that can be updated without altering values in any
87 of the parent mappings.
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +000088
Vinay Sajip1ba81ee2013-01-11 23:39:53 +000089 .. versionchanged:: 3.4
90 The optional ``m`` parameter was added.
91
Raymond Hettinger2a61c452012-07-15 22:37:20 -070092 .. attribute:: parents
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +000093
Raymond Hettingerb22ba042012-07-16 02:07:41 -070094 Property returning a new :class:`ChainMap` containing all of the maps in
Raymond Hettinger2a61c452012-07-15 22:37:20 -070095 the current instance except the first one. This is useful for skipping
96 the first map in the search. Use cases are similar to those for the
97 :keyword:`nonlocal` keyword used in :term:`nested scopes <nested
98 scope>`. The use cases also parallel those for the built-in
99 :func:`super` function. A reference to ``d.parents`` is equivalent to:
100 ``ChainMap(*d.maps[1:])``.
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +0000101
Raymond Hettingerbfcb4292012-06-10 11:39:44 -0700102
103.. seealso::
104
105 * The `MultiContext class
Sandro Tosiea475302012-08-12 10:37:23 +0200106 <https://github.com/enthought/codetools/blob/4.0.0/codetools/contexts/multi_context.py>`_
Raymond Hettingerbfcb4292012-06-10 11:39:44 -0700107 in the Enthought `CodeTools package
108 <https://github.com/enthought/codetools>`_ has options to support
109 writing to any mapping in the chain.
110
111 * Django's `Context class
Georg Brandl525d3552014-10-29 10:26:56 +0100112 <https://github.com/django/django/blob/master/django/template/context.py>`_
Raymond Hettingerbfcb4292012-06-10 11:39:44 -0700113 for templating is a read-only chain of mappings. It also features
114 pushing and popping of contexts similar to the
115 :meth:`~collections.ChainMap.new_child` method and the
116 :meth:`~collections.ChainMap.parents` property.
117
118 * The `Nested Contexts recipe
119 <http://code.activestate.com/recipes/577434/>`_ has options to control
120 whether writes and other mutations apply only to the first mapping or to
121 any mapping in the chain.
122
123 * A `greatly simplified read-only version of Chainmap
124 <http://code.activestate.com/recipes/305268/>`_.
125
126
127:class:`ChainMap` Examples and Recipes
128^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
129
130This section shows various approaches to working with chained maps.
131
132
133Example of simulating Python's internal lookup chain::
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +0000134
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700135 import builtins
136 pylookup = ChainMap(locals(), globals(), vars(builtins))
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +0000137
Raymond Hettingerb2269ba2012-07-15 23:53:32 -0700138Example of letting user specified command-line arguments take precedence over
139environment variables which in turn take precedence over default values::
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +0000140
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700141 import os, argparse
Raymond Hettingerb2269ba2012-07-15 23:53:32 -0700142
143 defaults = {'color': 'red', 'user': 'guest'}
144
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700145 parser = argparse.ArgumentParser()
146 parser.add_argument('-u', '--user')
147 parser.add_argument('-c', '--color')
Raymond Hettingerb2269ba2012-07-15 23:53:32 -0700148 namespace = parser.parse_args()
149 command_line_args = {k:v for k, v in vars(namespace).items() if v}
150
151 combined = ChainMap(command_line_args, os.environ, defaults)
152 print(combined['color'])
153 print(combined['user'])
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +0000154
Raymond Hettingerbfcb4292012-06-10 11:39:44 -0700155Example patterns for using the :class:`ChainMap` class to simulate nested
156contexts::
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +0000157
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700158 c = ChainMap() # Create root context
159 d = c.new_child() # Create nested child context
160 e = c.new_child() # Child of c, independent from d
161 e.maps[0] # Current context dictionary -- like Python's locals()
162 e.maps[-1] # Root context -- like Python's globals()
163 e.parents # Enclosing context chain -- like Python's nonlocals
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +0000164
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700165 d['x'] # Get first key in the chain of contexts
166 d['x'] = 1 # Set value in current context
Andrew Svetlov1a8db9c2012-10-04 19:29:25 +0300167 del d['x'] # Delete from current context
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700168 list(d) # All nested values
169 k in d # Check all nested values
170 len(d) # Number of nested values
171 d.items() # All nested items
172 dict(d) # Flatten into a regular dictionary
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +0000173
Raymond Hettingerbfcb4292012-06-10 11:39:44 -0700174The :class:`ChainMap` class only makes updates (writes and deletions) to the
175first mapping in the chain while lookups will search the full chain. However,
176if deep writes and deletions are desired, it is easy to make a subclass that
177updates keys found deeper in the chain::
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +0000178
Raymond Hettingerbfcb4292012-06-10 11:39:44 -0700179 class DeepChainMap(ChainMap):
180 'Variant of ChainMap that allows direct updates to inner scopes'
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +0000181
Raymond Hettingerbfcb4292012-06-10 11:39:44 -0700182 def __setitem__(self, key, value):
183 for mapping in self.maps:
184 if key in mapping:
185 mapping[key] = value
186 return
187 self.maps[0][key] = value
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +0000188
Raymond Hettingerbfcb4292012-06-10 11:39:44 -0700189 def __delitem__(self, key):
190 for mapping in self.maps:
191 if key in mapping:
192 del mapping[key]
193 return
194 raise KeyError(key)
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +0000195
Serhiy Storchakaf47036c2013-12-24 11:04:36 +0200196 >>> d = DeepChainMap({'zebra': 'black'}, {'elephant': 'blue'}, {'lion': 'yellow'})
Raymond Hettingerbfcb4292012-06-10 11:39:44 -0700197 >>> d['lion'] = 'orange' # update an existing key two levels down
198 >>> d['snake'] = 'red' # new keys get added to the topmost dict
199 >>> del d['elephant'] # remove an existing key one level down
200 DeepChainMap({'zebra': 'black', 'snake': 'red'}, {}, {'lion': 'orange'})
Georg Brandl4dcf4742012-03-08 20:35:08 +0100201
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +0000202
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000203:class:`Counter` objects
204------------------------
205
206A counter tool is provided to support convenient and rapid tallies.
207For example::
208
Raymond Hettinger1c62dc92009-02-04 11:41:45 +0000209 >>> # Tally occurrences of words in a list
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000210 >>> cnt = Counter()
Raymond Hettinger670eaec2009-01-21 23:14:07 +0000211 >>> for word in ['red', 'blue', 'red', 'green', 'blue', 'blue']:
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000212 ... cnt[word] += 1
213 >>> cnt
214 Counter({'blue': 3, 'red': 2, 'green': 1})
215
Raymond Hettinger1c62dc92009-02-04 11:41:45 +0000216 >>> # Find the ten most common words in Hamlet
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000217 >>> import re
Raymond Hettingerfaaba592013-03-01 03:30:20 -0800218 >>> words = re.findall(r'\w+', open('hamlet.txt').read().lower())
Raymond Hettinger0bae6622009-01-20 13:00:59 +0000219 >>> Counter(words).most_common(10)
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000220 [('the', 1143), ('and', 966), ('to', 762), ('of', 669), ('i', 631),
221 ('you', 554), ('a', 546), ('my', 514), ('hamlet', 471), ('in', 451)]
222
223.. class:: Counter([iterable-or-mapping])
224
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700225 A :class:`Counter` is a :class:`dict` subclass for counting hashable objects.
226 It is an unordered collection where elements are stored as dictionary keys
227 and their counts are stored as dictionary values. Counts are allowed to be
228 any integer value including zero or negative counts. The :class:`Counter`
229 class is similar to bags or multisets in other languages.
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000230
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700231 Elements are counted from an *iterable* or initialized from another
232 *mapping* (or counter):
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000233
Raymond Hettinger73662a52009-01-27 02:38:22 +0000234 >>> c = Counter() # a new, empty counter
235 >>> c = Counter('gallahad') # a new counter from an iterable
236 >>> c = Counter({'red': 4, 'blue': 2}) # a new counter from a mapping
237 >>> c = Counter(cats=4, dogs=8) # a new counter from keyword args
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000238
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700239 Counter objects have a dictionary interface except that they return a zero
240 count for missing items instead of raising a :exc:`KeyError`:
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000241
Raymond Hettinger94adc8e2009-01-22 05:27:37 +0000242 >>> c = Counter(['eggs', 'ham'])
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000243 >>> c['bacon'] # count of a missing element is zero
244 0
245
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700246 Setting a count to zero does not remove an element from a counter.
247 Use ``del`` to remove it entirely:
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000248
Raymond Hettinger94adc8e2009-01-22 05:27:37 +0000249 >>> c['sausage'] = 0 # counter entry with a zero count
250 >>> del c['sausage'] # del actually removes the entry
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000251
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700252 .. versionadded:: 3.1
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000253
254
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700255 Counter objects support three methods beyond those available for all
256 dictionaries:
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000257
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700258 .. method:: elements()
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000259
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700260 Return an iterator over elements repeating each as many times as its
261 count. Elements are returned in arbitrary order. If an element's count
262 is less than one, :meth:`elements` will ignore it.
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000263
Raymond Hettinger0bae6622009-01-20 13:00:59 +0000264 >>> c = Counter(a=4, b=2, c=0, d=-2)
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000265 >>> list(c.elements())
266 ['a', 'a', 'a', 'a', 'b', 'b']
267
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700268 .. method:: most_common([n])
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000269
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700270 Return a list of the *n* most common elements and their counts from the
Raymond Hettingera3780252015-05-13 02:47:57 -0700271 most common to the least. If *n* is omitted or ``None``,
Raymond Hettinger3afdb282015-05-13 14:39:04 -0700272 :func:`most_common` returns *all* elements in the counter.
273 Elements with equal counts are ordered arbitrarily:
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000274
275 >>> Counter('abracadabra').most_common(3)
276 [('a', 5), ('r', 2), ('b', 2)]
277
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700278 .. method:: subtract([iterable-or-mapping])
Raymond Hettinger9c01e442010-04-03 10:32:58 +0000279
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700280 Elements are subtracted from an *iterable* or from another *mapping*
281 (or counter). Like :meth:`dict.update` but subtracts counts instead
282 of replacing them. Both inputs and outputs may be zero or negative.
Raymond Hettinger9c01e442010-04-03 10:32:58 +0000283
284 >>> c = Counter(a=4, b=2, c=0, d=-2)
285 >>> d = Counter(a=1, b=2, c=3, d=4)
286 >>> c.subtract(d)
Andrew Svetlovf6351722012-12-17 14:01:16 +0200287 >>> c
Raymond Hettinger9c01e442010-04-03 10:32:58 +0000288 Counter({'a': 3, 'b': 0, 'c': -3, 'd': -6})
289
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700290 .. versionadded:: 3.2
Ezio Melotti0be8b1c2010-04-04 06:53:44 +0000291
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700292 The usual dictionary methods are available for :class:`Counter` objects
293 except for two which work differently for counters.
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000294
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700295 .. method:: fromkeys(iterable)
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000296
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700297 This class method is not implemented for :class:`Counter` objects.
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000298
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700299 .. method:: update([iterable-or-mapping])
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000300
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700301 Elements are counted from an *iterable* or added-in from another
302 *mapping* (or counter). Like :meth:`dict.update` but adds counts
303 instead of replacing them. Also, the *iterable* is expected to be a
304 sequence of elements, not a sequence of ``(key, value)`` pairs.
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000305
306Common patterns for working with :class:`Counter` objects::
307
Raymond Hettinger73662a52009-01-27 02:38:22 +0000308 sum(c.values()) # total of all counts
309 c.clear() # reset all counts
310 list(c) # list unique elements
311 set(c) # convert to a set
312 dict(c) # convert to a regular dictionary
313 c.items() # convert to a list of (elem, cnt) pairs
314 Counter(dict(list_of_pairs)) # convert from a list of (elem, cnt) pairs
Georg Brandl87f3d7b2013-10-06 12:36:39 +0200315 c.most_common()[:-n-1:-1] # n least common elements
Raymond Hettingerfcb393c2011-08-09 13:00:40 -0700316 +c # remove zero and negative counts
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000317
Raymond Hettinger72a95cc2009-02-25 22:51:40 +0000318Several mathematical operations are provided for combining :class:`Counter`
319objects to produce multisets (counters that have counts greater than zero).
320Addition and subtraction combine counters by adding or subtracting the counts
321of corresponding elements. Intersection and union return the minimum and
322maximum of corresponding counts. Each operation can accept inputs with signed
323counts, but the output will exclude results with counts of zero or less.
Raymond Hettinger4d2073a2009-01-20 03:41:22 +0000324
Raymond Hettingere0d1b9f2009-01-21 20:36:27 +0000325 >>> c = Counter(a=3, b=1)
326 >>> d = Counter(a=1, b=2)
Raymond Hettinger73662a52009-01-27 02:38:22 +0000327 >>> c + d # add two counters together: c[x] + d[x]
Raymond Hettinger4d2073a2009-01-20 03:41:22 +0000328 Counter({'a': 4, 'b': 3})
Raymond Hettinger73662a52009-01-27 02:38:22 +0000329 >>> c - d # subtract (keeping only positive counts)
Raymond Hettinger4d2073a2009-01-20 03:41:22 +0000330 Counter({'a': 2})
Raymond Hettinger73662a52009-01-27 02:38:22 +0000331 >>> c & d # intersection: min(c[x], d[x])
Raymond Hettinger4d2073a2009-01-20 03:41:22 +0000332 Counter({'a': 1, 'b': 1})
Raymond Hettinger73662a52009-01-27 02:38:22 +0000333 >>> c | d # union: max(c[x], d[x])
Raymond Hettinger4d2073a2009-01-20 03:41:22 +0000334 Counter({'a': 3, 'b': 2})
335
Berker Peksag315e1042015-05-19 01:36:55 +0300336Unary addition and subtraction are shortcuts for adding an empty counter
Raymond Hettingerfcb393c2011-08-09 13:00:40 -0700337or subtracting from an empty counter.
338
339 >>> c = Counter(a=2, b=-4)
340 >>> +c
341 Counter({'a': 2})
342 >>> -c
343 Counter({'b': 4})
344
345.. versionadded:: 3.3
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700346 Added support for unary plus, unary minus, and in-place multiset operations.
Raymond Hettingerfcb393c2011-08-09 13:00:40 -0700347
Raymond Hettinger22f18852010-04-12 21:45:14 +0000348.. note::
349
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700350 Counters were primarily designed to work with positive integers to represent
351 running counts; however, care was taken to not unnecessarily preclude use
352 cases needing other types or negative values. To help with those use cases,
353 this section documents the minimum range and type restrictions.
Raymond Hettinger22f18852010-04-12 21:45:14 +0000354
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700355 * The :class:`Counter` class itself is a dictionary subclass with no
Georg Brandl2fdc0f82012-10-06 22:38:20 +0200356 restrictions on its keys and values. The values are intended to be numbers
357 representing counts, but you *could* store anything in the value field.
Raymond Hettinger22f18852010-04-12 21:45:14 +0000358
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700359 * The :meth:`most_common` method requires only that the values be orderable.
Raymond Hettinger22f18852010-04-12 21:45:14 +0000360
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700361 * For in-place operations such as ``c[key] += 1``, the value type need only
Georg Brandl2fdc0f82012-10-06 22:38:20 +0200362 support addition and subtraction. So fractions, floats, and decimals would
363 work and negative values are supported. The same is also true for
364 :meth:`update` and :meth:`subtract` which allow negative and zero values
365 for both inputs and outputs.
Raymond Hettinger22f18852010-04-12 21:45:14 +0000366
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700367 * The multiset methods are designed only for use cases with positive values.
Georg Brandl2fdc0f82012-10-06 22:38:20 +0200368 The inputs may be negative or zero, but only outputs with positive values
369 are created. There are no type restrictions, but the value type needs to
370 support addition, subtraction, and comparison.
Raymond Hettinger22f18852010-04-12 21:45:14 +0000371
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700372 * The :meth:`elements` method requires integer counts. It ignores zero and
Georg Brandl2fdc0f82012-10-06 22:38:20 +0200373 negative counts.
Raymond Hettinger22f18852010-04-12 21:45:14 +0000374
Raymond Hettingerb14043c2009-01-20 23:44:31 +0000375.. seealso::
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000376
Raymond Hettingerb14043c2009-01-20 23:44:31 +0000377 * `Bag class <http://www.gnu.org/software/smalltalk/manual-base/html_node/Bag.html>`_
Raymond Hettingerbfcb4292012-06-10 11:39:44 -0700378 in Smalltalk.
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000379
Éric Araujo08c9bd52011-04-24 02:59:02 +0200380 * Wikipedia entry for `Multisets <http://en.wikipedia.org/wiki/Multiset>`_.
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000381
Raymond Hettingerb14043c2009-01-20 23:44:31 +0000382 * `C++ multisets <http://www.demo2s.com/Tutorial/Cpp/0380__set-multiset/Catalog0380__set-multiset.htm>`_
Raymond Hettingerbfcb4292012-06-10 11:39:44 -0700383 tutorial with examples.
Raymond Hettingerb14043c2009-01-20 23:44:31 +0000384
Raymond Hettinger94adc8e2009-01-22 05:27:37 +0000385 * For mathematical operations on multisets and their use cases, see
Raymond Hettingerbfcb4292012-06-10 11:39:44 -0700386 *Knuth, Donald. The Art of Computer Programming Volume II,
387 Section 4.6.3, Exercise 19*.
Raymond Hettingerb14043c2009-01-20 23:44:31 +0000388
Raymond Hettinger670eaec2009-01-21 23:14:07 +0000389 * To enumerate all distinct multisets of a given size over a given set of
Raymond Hettinger855482e2015-05-23 08:57:58 -0700390 elements, see :func:`itertools.combinations_with_replacement`:
Raymond Hettingerb14043c2009-01-20 23:44:31 +0000391
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700392 map(Counter, combinations_with_replacement('ABC', 2)) --> AA AB AC BB BC CC
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000393
394
Georg Brandl116aa622007-08-15 14:28:22 +0000395:class:`deque` objects
396----------------------
397
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000398.. class:: deque([iterable, [maxlen]])
Georg Brandl116aa622007-08-15 14:28:22 +0000399
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700400 Returns a new deque object initialized left-to-right (using :meth:`append`) with
401 data from *iterable*. If *iterable* is not specified, the new deque is empty.
Georg Brandl116aa622007-08-15 14:28:22 +0000402
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700403 Deques are a generalization of stacks and queues (the name is pronounced "deck"
404 and is short for "double-ended queue"). Deques support thread-safe, memory
405 efficient appends and pops from either side of the deque with approximately the
406 same O(1) performance in either direction.
Georg Brandl116aa622007-08-15 14:28:22 +0000407
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700408 Though :class:`list` objects support similar operations, they are optimized for
409 fast fixed-length operations and incur O(n) memory movement costs for
410 ``pop(0)`` and ``insert(0, v)`` operations which change both the size and
411 position of the underlying data representation.
Georg Brandl116aa622007-08-15 14:28:22 +0000412
Georg Brandl116aa622007-08-15 14:28:22 +0000413
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700414 If *maxlen* is not specified or is *None*, deques may grow to an
415 arbitrary length. Otherwise, the deque is bounded to the specified maximum
416 length. Once a bounded length deque is full, when new items are added, a
417 corresponding number of items are discarded from the opposite end. Bounded
418 length deques provide functionality similar to the ``tail`` filter in
419 Unix. They are also useful for tracking transactions and other pools of data
420 where only the most recent activity is of interest.
Georg Brandl9afde1c2007-11-01 20:32:30 +0000421
Georg Brandl9afde1c2007-11-01 20:32:30 +0000422
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700423 Deque objects support the following methods:
Georg Brandl116aa622007-08-15 14:28:22 +0000424
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700425 .. method:: append(x)
Georg Brandl116aa622007-08-15 14:28:22 +0000426
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700427 Add *x* to the right side of the deque.
Georg Brandl116aa622007-08-15 14:28:22 +0000428
429
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700430 .. method:: appendleft(x)
Georg Brandl116aa622007-08-15 14:28:22 +0000431
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700432 Add *x* to the left side of the deque.
Georg Brandl116aa622007-08-15 14:28:22 +0000433
434
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700435 .. method:: clear()
Georg Brandl116aa622007-08-15 14:28:22 +0000436
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700437 Remove all elements from the deque leaving it with length 0.
Georg Brandl116aa622007-08-15 14:28:22 +0000438
439
Raymond Hettinger32ea1652015-03-21 01:37:37 -0700440 .. method:: copy()
441
442 Create a shallow copy of the deque.
443
444 .. versionadded:: 3.5
445
446
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700447 .. method:: count(x)
Raymond Hettinger44459de2010-04-03 23:20:46 +0000448
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700449 Count the number of deque elements equal to *x*.
Raymond Hettinger44459de2010-04-03 23:20:46 +0000450
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700451 .. versionadded:: 3.2
Raymond Hettinger44459de2010-04-03 23:20:46 +0000452
Georg Brandl67b21b72010-08-17 15:07:14 +0000453
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700454 .. method:: extend(iterable)
Georg Brandl116aa622007-08-15 14:28:22 +0000455
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700456 Extend the right side of the deque by appending elements from the iterable
457 argument.
Georg Brandl116aa622007-08-15 14:28:22 +0000458
459
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700460 .. method:: extendleft(iterable)
Georg Brandl116aa622007-08-15 14:28:22 +0000461
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700462 Extend the left side of the deque by appending elements from *iterable*.
463 Note, the series of left appends results in reversing the order of
464 elements in the iterable argument.
Georg Brandl116aa622007-08-15 14:28:22 +0000465
466
Raymond Hettinger855482e2015-05-23 08:57:58 -0700467 .. method:: index(x[, start[, stop]])
Raymond Hettinger32ea1652015-03-21 01:37:37 -0700468
Raymond Hettinger855482e2015-05-23 08:57:58 -0700469 Return the position of *x* in the deque (at or after index *start*
470 and before index *stop*). Returns the first match or raises
471 :exc:`ValueError` if not found.
Raymond Hettinger32ea1652015-03-21 01:37:37 -0700472
473 .. versionadded:: 3.5
474
475
476 .. method:: insert(i, x)
477
478 Insert *x* into the deque at position *i*.
479
480 .. versionadded:: 3.5
481
482
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700483 .. method:: pop()
Georg Brandl116aa622007-08-15 14:28:22 +0000484
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700485 Remove and return an element from the right side of the deque. If no
486 elements are present, raises an :exc:`IndexError`.
Georg Brandl116aa622007-08-15 14:28:22 +0000487
488
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700489 .. method:: popleft()
Georg Brandl116aa622007-08-15 14:28:22 +0000490
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700491 Remove and return an element from the left side of the deque. If no
492 elements are present, raises an :exc:`IndexError`.
Georg Brandl116aa622007-08-15 14:28:22 +0000493
494
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700495 .. method:: remove(value)
Georg Brandl116aa622007-08-15 14:28:22 +0000496
Raymond Hettinger855482e2015-05-23 08:57:58 -0700497 Remove the first occurrence of *value*. If not found, raises a
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700498 :exc:`ValueError`.
Georg Brandl116aa622007-08-15 14:28:22 +0000499
Georg Brandl67b21b72010-08-17 15:07:14 +0000500
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700501 .. method:: reverse()
Raymond Hettingere5fdedb2009-12-10 00:47:21 +0000502
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700503 Reverse the elements of the deque in-place and then return ``None``.
Raymond Hettingere5fdedb2009-12-10 00:47:21 +0000504
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700505 .. versionadded:: 3.2
Georg Brandl116aa622007-08-15 14:28:22 +0000506
Georg Brandl67b21b72010-08-17 15:07:14 +0000507
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700508 .. method:: rotate(n)
Georg Brandl116aa622007-08-15 14:28:22 +0000509
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700510 Rotate the deque *n* steps to the right. If *n* is negative, rotate to
511 the left. Rotating one step to the right is equivalent to:
512 ``d.appendleft(d.pop())``.
Benjamin Petersone41251e2008-04-25 01:59:09 +0000513
Georg Brandl116aa622007-08-15 14:28:22 +0000514
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700515 Deque objects also provide one read-only attribute:
Raymond Hettinger5bb0f0e2009-03-10 12:56:32 +0000516
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700517 .. attribute:: maxlen
Raymond Hettinger5bb0f0e2009-03-10 12:56:32 +0000518
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700519 Maximum size of a deque or *None* if unbounded.
Raymond Hettinger5bb0f0e2009-03-10 12:56:32 +0000520
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700521 .. versionadded:: 3.1
Raymond Hettinger5bb0f0e2009-03-10 12:56:32 +0000522
523
Georg Brandl116aa622007-08-15 14:28:22 +0000524In addition to the above, deques support iteration, pickling, ``len(d)``,
525``reversed(d)``, ``copy.copy(d)``, ``copy.deepcopy(d)``, membership testing with
Benjamin Peterson206e3072008-10-19 14:07:49 +0000526the :keyword:`in` operator, and subscript references such as ``d[-1]``. Indexed
527access is O(1) at both ends but slows to O(n) in the middle. For fast random
528access, use lists instead.
Georg Brandl116aa622007-08-15 14:28:22 +0000529
Raymond Hettinger41290a62015-03-31 08:12:23 -0700530Starting in version 3.5, deques support ``__add__()``, ``__mul__()``,
531and ``__imul__()``.
532
Christian Heimesfe337bf2008-03-23 21:54:12 +0000533Example:
534
535.. doctest::
Georg Brandl116aa622007-08-15 14:28:22 +0000536
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700537 >>> from collections import deque
538 >>> d = deque('ghi') # make a new deque with three items
539 >>> for elem in d: # iterate over the deque's elements
540 ... print(elem.upper())
541 G
542 H
543 I
Georg Brandl116aa622007-08-15 14:28:22 +0000544
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700545 >>> d.append('j') # add a new entry to the right side
546 >>> d.appendleft('f') # add a new entry to the left side
547 >>> d # show the representation of the deque
548 deque(['f', 'g', 'h', 'i', 'j'])
Georg Brandl116aa622007-08-15 14:28:22 +0000549
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700550 >>> d.pop() # return and remove the rightmost item
551 'j'
552 >>> d.popleft() # return and remove the leftmost item
553 'f'
554 >>> list(d) # list the contents of the deque
555 ['g', 'h', 'i']
556 >>> d[0] # peek at leftmost item
557 'g'
558 >>> d[-1] # peek at rightmost item
559 'i'
Georg Brandl116aa622007-08-15 14:28:22 +0000560
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700561 >>> list(reversed(d)) # list the contents of a deque in reverse
562 ['i', 'h', 'g']
563 >>> 'h' in d # search the deque
564 True
565 >>> d.extend('jkl') # add multiple elements at once
566 >>> d
567 deque(['g', 'h', 'i', 'j', 'k', 'l'])
568 >>> d.rotate(1) # right rotation
569 >>> d
570 deque(['l', 'g', 'h', 'i', 'j', 'k'])
571 >>> d.rotate(-1) # left rotation
572 >>> d
573 deque(['g', 'h', 'i', 'j', 'k', 'l'])
Georg Brandl116aa622007-08-15 14:28:22 +0000574
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700575 >>> deque(reversed(d)) # make a new deque in reverse order
576 deque(['l', 'k', 'j', 'i', 'h', 'g'])
577 >>> d.clear() # empty the deque
578 >>> d.pop() # cannot pop from an empty deque
579 Traceback (most recent call last):
580 File "<pyshell#6>", line 1, in -toplevel-
581 d.pop()
582 IndexError: pop from an empty deque
Georg Brandl116aa622007-08-15 14:28:22 +0000583
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700584 >>> d.extendleft('abc') # extendleft() reverses the input order
585 >>> d
586 deque(['c', 'b', 'a'])
Georg Brandl116aa622007-08-15 14:28:22 +0000587
588
Georg Brandl9afde1c2007-11-01 20:32:30 +0000589:class:`deque` Recipes
590^^^^^^^^^^^^^^^^^^^^^^
Georg Brandl116aa622007-08-15 14:28:22 +0000591
592This section shows various approaches to working with deques.
593
Raymond Hettingerd2ee64d2009-03-31 22:52:48 +0000594Bounded length deques provide functionality similar to the ``tail`` filter
595in Unix::
596
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700597 def tail(filename, n=10):
598 'Return the last n lines of a file'
599 with open(filename) as f:
600 return deque(f, n)
Raymond Hettingerd2ee64d2009-03-31 22:52:48 +0000601
602Another approach to using deques is to maintain a sequence of recently
603added elements by appending to the right and popping to the left::
604
605 def moving_average(iterable, n=3):
606 # moving_average([40, 30, 50, 46, 39, 44]) --> 40.0 42.0 45.0 43.0
607 # http://en.wikipedia.org/wiki/Moving_average
608 it = iter(iterable)
Raymond Hettingerd40285a2009-05-22 01:11:26 +0000609 d = deque(itertools.islice(it, n-1))
610 d.appendleft(0)
Raymond Hettingerd2ee64d2009-03-31 22:52:48 +0000611 s = sum(d)
Raymond Hettingerd2ee64d2009-03-31 22:52:48 +0000612 for elem in it:
613 s += elem - d.popleft()
614 d.append(elem)
615 yield s / n
616
Georg Brandl116aa622007-08-15 14:28:22 +0000617The :meth:`rotate` method provides a way to implement :class:`deque` slicing and
Ezio Melotti0639d5a2009-12-19 23:26:38 +0000618deletion. For example, a pure Python implementation of ``del d[n]`` relies on
Georg Brandl116aa622007-08-15 14:28:22 +0000619the :meth:`rotate` method to position elements to be popped::
620
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700621 def delete_nth(d, n):
622 d.rotate(-n)
623 d.popleft()
624 d.rotate(n)
Georg Brandl116aa622007-08-15 14:28:22 +0000625
626To implement :class:`deque` slicing, use a similar approach applying
627:meth:`rotate` to bring a target element to the left side of the deque. Remove
628old entries with :meth:`popleft`, add new entries with :meth:`extend`, and then
629reverse the rotation.
Georg Brandl116aa622007-08-15 14:28:22 +0000630With minor variations on that approach, it is easy to implement Forth style
631stack manipulations such as ``dup``, ``drop``, ``swap``, ``over``, ``pick``,
632``rot``, and ``roll``.
633
Georg Brandl116aa622007-08-15 14:28:22 +0000634
635:class:`defaultdict` objects
636----------------------------
637
Georg Brandl116aa622007-08-15 14:28:22 +0000638.. class:: defaultdict([default_factory[, ...]])
639
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700640 Returns a new dictionary-like object. :class:`defaultdict` is a subclass of the
641 built-in :class:`dict` class. It overrides one method and adds one writable
642 instance variable. The remaining functionality is the same as for the
643 :class:`dict` class and is not documented here.
Georg Brandl116aa622007-08-15 14:28:22 +0000644
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700645 The first argument provides the initial value for the :attr:`default_factory`
646 attribute; it defaults to ``None``. All remaining arguments are treated the same
647 as if they were passed to the :class:`dict` constructor, including keyword
648 arguments.
Georg Brandl116aa622007-08-15 14:28:22 +0000649
Georg Brandl116aa622007-08-15 14:28:22 +0000650
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700651 :class:`defaultdict` objects support the following method in addition to the
652 standard :class:`dict` operations:
Georg Brandl116aa622007-08-15 14:28:22 +0000653
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700654 .. method:: __missing__(key)
Georg Brandl116aa622007-08-15 14:28:22 +0000655
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700656 If the :attr:`default_factory` attribute is ``None``, this raises a
657 :exc:`KeyError` exception with the *key* as argument.
Georg Brandl116aa622007-08-15 14:28:22 +0000658
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700659 If :attr:`default_factory` is not ``None``, it is called without arguments
660 to provide a default value for the given *key*, this value is inserted in
661 the dictionary for the *key*, and returned.
Georg Brandl116aa622007-08-15 14:28:22 +0000662
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700663 If calling :attr:`default_factory` raises an exception this exception is
664 propagated unchanged.
Georg Brandl116aa622007-08-15 14:28:22 +0000665
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700666 This method is called by the :meth:`__getitem__` method of the
667 :class:`dict` class when the requested key is not found; whatever it
668 returns or raises is then returned or raised by :meth:`__getitem__`.
Georg Brandl116aa622007-08-15 14:28:22 +0000669
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700670 Note that :meth:`__missing__` is *not* called for any operations besides
671 :meth:`__getitem__`. This means that :meth:`get` will, like normal
672 dictionaries, return ``None`` as a default rather than using
673 :attr:`default_factory`.
Benjamin Peterson871b9d12012-01-27 09:14:01 -0500674
Georg Brandl116aa622007-08-15 14:28:22 +0000675
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700676 :class:`defaultdict` objects support the following instance variable:
Georg Brandl116aa622007-08-15 14:28:22 +0000677
Benjamin Petersone41251e2008-04-25 01:59:09 +0000678
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700679 .. attribute:: default_factory
Benjamin Petersone41251e2008-04-25 01:59:09 +0000680
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700681 This attribute is used by the :meth:`__missing__` method; it is
682 initialized from the first argument to the constructor, if present, or to
683 ``None``, if absent.
Georg Brandl116aa622007-08-15 14:28:22 +0000684
685
Georg Brandl116aa622007-08-15 14:28:22 +0000686:class:`defaultdict` Examples
687^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
688
689Using :class:`list` as the :attr:`default_factory`, it is easy to group a
Christian Heimesfe337bf2008-03-23 21:54:12 +0000690sequence of key-value pairs into a dictionary of lists:
Georg Brandl116aa622007-08-15 14:28:22 +0000691
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700692 >>> s = [('yellow', 1), ('blue', 2), ('yellow', 3), ('blue', 4), ('red', 1)]
693 >>> d = defaultdict(list)
694 >>> for k, v in s:
695 ... d[k].append(v)
696 ...
697 >>> list(d.items())
698 [('blue', [2, 4]), ('red', [1]), ('yellow', [1, 3])]
Georg Brandl116aa622007-08-15 14:28:22 +0000699
700When each key is encountered for the first time, it is not already in the
701mapping; so an entry is automatically created using the :attr:`default_factory`
702function which returns an empty :class:`list`. The :meth:`list.append`
703operation then attaches the value to the new list. When keys are encountered
704again, the look-up proceeds normally (returning the list for that key) and the
705:meth:`list.append` operation adds another value to the list. This technique is
Christian Heimesfe337bf2008-03-23 21:54:12 +0000706simpler and faster than an equivalent technique using :meth:`dict.setdefault`:
Georg Brandl116aa622007-08-15 14:28:22 +0000707
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700708 >>> d = {}
709 >>> for k, v in s:
710 ... d.setdefault(k, []).append(v)
711 ...
712 >>> list(d.items())
713 [('blue', [2, 4]), ('red', [1]), ('yellow', [1, 3])]
Georg Brandl116aa622007-08-15 14:28:22 +0000714
715Setting the :attr:`default_factory` to :class:`int` makes the
716:class:`defaultdict` useful for counting (like a bag or multiset in other
Christian Heimesfe337bf2008-03-23 21:54:12 +0000717languages):
Georg Brandl116aa622007-08-15 14:28:22 +0000718
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700719 >>> s = 'mississippi'
720 >>> d = defaultdict(int)
721 >>> for k in s:
722 ... d[k] += 1
723 ...
724 >>> list(d.items())
725 [('i', 4), ('p', 2), ('s', 4), ('m', 1)]
Georg Brandl116aa622007-08-15 14:28:22 +0000726
727When a letter is first encountered, it is missing from the mapping, so the
728:attr:`default_factory` function calls :func:`int` to supply a default count of
729zero. The increment operation then builds up the count for each letter.
730
731The function :func:`int` which always returns zero is just a special case of
732constant functions. A faster and more flexible way to create constant functions
733is to use a lambda function which can supply any constant value (not just
Christian Heimesfe337bf2008-03-23 21:54:12 +0000734zero):
Georg Brandl116aa622007-08-15 14:28:22 +0000735
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700736 >>> def constant_factory(value):
737 ... return lambda: value
738 >>> d = defaultdict(constant_factory('<missing>'))
739 >>> d.update(name='John', action='ran')
740 >>> '%(name)s %(action)s to %(object)s' % d
741 'John ran to <missing>'
Georg Brandl116aa622007-08-15 14:28:22 +0000742
743Setting the :attr:`default_factory` to :class:`set` makes the
Christian Heimesfe337bf2008-03-23 21:54:12 +0000744:class:`defaultdict` useful for building a dictionary of sets:
Georg Brandl116aa622007-08-15 14:28:22 +0000745
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700746 >>> s = [('red', 1), ('blue', 2), ('red', 3), ('blue', 4), ('red', 1), ('blue', 4)]
747 >>> d = defaultdict(set)
748 >>> for k, v in s:
749 ... d[k].add(v)
750 ...
751 >>> list(d.items())
752 [('blue', {2, 4}), ('red', {1, 3})]
Georg Brandl116aa622007-08-15 14:28:22 +0000753
754
Guido van Rossum3d392eb2007-11-16 00:35:22 +0000755:func:`namedtuple` Factory Function for Tuples with Named Fields
Christian Heimes790c8232008-01-07 21:14:23 +0000756----------------------------------------------------------------
Georg Brandl116aa622007-08-15 14:28:22 +0000757
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000758Named tuples assign meaning to each position in a tuple and allow for more readable,
759self-documenting code. They can be used wherever regular tuples are used, and
760they add the ability to access fields by name instead of position index.
Georg Brandl116aa622007-08-15 14:28:22 +0000761
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000762.. function:: namedtuple(typename, field_names, verbose=False, rename=False)
Georg Brandl116aa622007-08-15 14:28:22 +0000763
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700764 Returns a new tuple subclass named *typename*. The new subclass is used to
765 create tuple-like objects that have fields accessible by attribute lookup as
766 well as being indexable and iterable. Instances of the subclass also have a
767 helpful docstring (with typename and field_names) and a helpful :meth:`__repr__`
768 method which lists the tuple contents in a ``name=value`` format.
Georg Brandl116aa622007-08-15 14:28:22 +0000769
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700770 The *field_names* are a single string with each fieldname separated by whitespace
771 and/or commas, for example ``'x y'`` or ``'x, y'``. Alternatively, *field_names*
772 can be a sequence of strings such as ``['x', 'y']``.
Georg Brandl9afde1c2007-11-01 20:32:30 +0000773
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700774 Any valid Python identifier may be used for a fieldname except for names
775 starting with an underscore. Valid identifiers consist of letters, digits,
776 and underscores but do not start with a digit or underscore and cannot be
777 a :mod:`keyword` such as *class*, *for*, *return*, *global*, *pass*,
778 or *raise*.
Georg Brandl116aa622007-08-15 14:28:22 +0000779
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700780 If *rename* is true, invalid fieldnames are automatically replaced
781 with positional names. For example, ``['abc', 'def', 'ghi', 'abc']`` is
782 converted to ``['abc', '_1', 'ghi', '_3']``, eliminating the keyword
783 ``def`` and the duplicate fieldname ``abc``.
Benjamin Petersona86f2c02009-02-10 02:41:10 +0000784
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700785 If *verbose* is true, the class definition is printed after it is
786 built. This option is outdated; instead, it is simpler to print the
787 :attr:`_source` attribute.
Georg Brandl116aa622007-08-15 14:28:22 +0000788
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700789 Named tuple instances do not have per-instance dictionaries, so they are
790 lightweight and require no more memory than regular tuples.
Georg Brandl116aa622007-08-15 14:28:22 +0000791
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700792 .. versionchanged:: 3.1
793 Added support for *rename*.
Benjamin Petersona86f2c02009-02-10 02:41:10 +0000794
Christian Heimesfe337bf2008-03-23 21:54:12 +0000795
796.. doctest::
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700797 :options: +NORMALIZE_WHITESPACE
Georg Brandl116aa622007-08-15 14:28:22 +0000798
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700799 >>> # Basic example
800 >>> Point = namedtuple('Point', ['x', 'y'])
801 >>> p = Point(11, y=22) # instantiate with positional or keyword arguments
802 >>> p[0] + p[1] # indexable like the plain tuple (11, 22)
803 33
804 >>> x, y = p # unpack like a regular tuple
805 >>> x, y
806 (11, 22)
807 >>> p.x + p.y # fields also accessible by name
808 33
809 >>> p # readable __repr__ with a name=value style
810 Point(x=11, y=22)
Georg Brandl116aa622007-08-15 14:28:22 +0000811
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000812Named tuples are especially useful for assigning field names to result tuples returned
813by the :mod:`csv` or :mod:`sqlite3` modules::
814
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700815 EmployeeRecord = namedtuple('EmployeeRecord', 'name, age, title, department, paygrade')
Georg Brandl9afde1c2007-11-01 20:32:30 +0000816
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700817 import csv
818 for emp in map(EmployeeRecord._make, csv.reader(open("employees.csv", "rb"))):
819 print(emp.name, emp.title)
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000820
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700821 import sqlite3
822 conn = sqlite3.connect('/companydata')
823 cursor = conn.cursor()
824 cursor.execute('SELECT name, age, title, department, paygrade FROM employees')
825 for emp in map(EmployeeRecord._make, cursor.fetchall()):
826 print(emp.name, emp.title)
Georg Brandl9afde1c2007-11-01 20:32:30 +0000827
Christian Heimes99170a52007-12-19 02:07:34 +0000828In addition to the methods inherited from tuples, named tuples support
Raymond Hettinger2ebea412011-03-23 12:52:23 -0700829three additional methods and two attributes. To prevent conflicts with
Christian Heimes2380ac72008-01-09 00:17:24 +0000830field names, the method and attribute names start with an underscore.
Christian Heimes99170a52007-12-19 02:07:34 +0000831
Benjamin Peterson0b9fb802010-07-18 14:23:36 +0000832.. classmethod:: somenamedtuple._make(iterable)
Christian Heimes99170a52007-12-19 02:07:34 +0000833
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700834 Class method that makes a new instance from an existing sequence or iterable.
Christian Heimes99170a52007-12-19 02:07:34 +0000835
Raymond Hettinger6fed9fd2012-06-11 00:38:14 -0700836 .. doctest::
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000837
Raymond Hettinger6fed9fd2012-06-11 00:38:14 -0700838 >>> t = [11, 22]
839 >>> Point._make(t)
840 Point(x=11, y=22)
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000841
Christian Heimes790c8232008-01-07 21:14:23 +0000842.. method:: somenamedtuple._asdict()
Georg Brandl9afde1c2007-11-01 20:32:30 +0000843
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700844 Return a new :class:`OrderedDict` which maps field names to their corresponding
845 values. Note, this method is no longer needed now that the same effect can
846 be achieved by using the built-in :func:`vars` function::
Georg Brandl9afde1c2007-11-01 20:32:30 +0000847
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700848 >>> vars(p)
849 OrderedDict([('x', 11), ('y', 22)])
Raymond Hettingera4f52b12009-03-02 22:28:31 +0000850
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700851 .. versionchanged:: 3.1
852 Returns an :class:`OrderedDict` instead of a regular :class:`dict`.
Christian Heimesfe337bf2008-03-23 21:54:12 +0000853
Christian Heimes790c8232008-01-07 21:14:23 +0000854.. method:: somenamedtuple._replace(kwargs)
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000855
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700856 Return a new instance of the named tuple replacing specified fields with new
Raymond Hettinger6fed9fd2012-06-11 00:38:14 -0700857 values::
Thomas Wouters8ce81f72007-09-20 18:22:40 +0000858
Raymond Hettinger6fed9fd2012-06-11 00:38:14 -0700859 >>> p = Point(x=11, y=22)
860 >>> p._replace(x=33)
861 Point(x=33, y=22)
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000862
Raymond Hettinger6fed9fd2012-06-11 00:38:14 -0700863 >>> for partnum, record in inventory.items():
864 ... inventory[partnum] = record._replace(price=newprices[partnum], timestamp=time.now())
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000865
Raymond Hettinger2ebea412011-03-23 12:52:23 -0700866.. attribute:: somenamedtuple._source
867
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700868 A string with the pure Python source code used to create the named
869 tuple class. The source makes the named tuple self-documenting.
Raymond Hettinger163e9822013-05-18 00:05:20 -0700870 It can be printed, executed using :func:`exec`, or saved to a file
871 and imported.
Raymond Hettinger2ebea412011-03-23 12:52:23 -0700872
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700873 .. versionadded:: 3.3
Raymond Hettinger2ebea412011-03-23 12:52:23 -0700874
Christian Heimes790c8232008-01-07 21:14:23 +0000875.. attribute:: somenamedtuple._fields
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000876
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700877 Tuple of strings listing the field names. Useful for introspection
878 and for creating new named tuple types from existing named tuples.
Thomas Wouters8ce81f72007-09-20 18:22:40 +0000879
Raymond Hettinger6fed9fd2012-06-11 00:38:14 -0700880 .. doctest::
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000881
Raymond Hettinger6fed9fd2012-06-11 00:38:14 -0700882 >>> p._fields # view the field names
883 ('x', 'y')
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000884
Raymond Hettinger6fed9fd2012-06-11 00:38:14 -0700885 >>> Color = namedtuple('Color', 'red green blue')
886 >>> Pixel = namedtuple('Pixel', Point._fields + Color._fields)
887 >>> Pixel(11, 22, 128, 255, 0)
888 Pixel(x=11, y=22, red=128, green=255, blue=0)
Georg Brandl116aa622007-08-15 14:28:22 +0000889
Christian Heimes0449f632007-12-15 01:27:15 +0000890To retrieve a field whose name is stored in a string, use the :func:`getattr`
Christian Heimesfe337bf2008-03-23 21:54:12 +0000891function:
Christian Heimes0449f632007-12-15 01:27:15 +0000892
893 >>> getattr(p, 'x')
894 11
895
Raymond Hettinger651453a2009-02-11 00:20:02 +0000896To convert a dictionary to a named tuple, use the double-star-operator
897(as described in :ref:`tut-unpacking-arguments`):
Christian Heimes99170a52007-12-19 02:07:34 +0000898
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700899 >>> d = {'x': 11, 'y': 22}
900 >>> Point(**d)
901 Point(x=11, y=22)
Christian Heimes99170a52007-12-19 02:07:34 +0000902
Guido van Rossum3d392eb2007-11-16 00:35:22 +0000903Since a named tuple is a regular Python class, it is easy to add or change
Christian Heimes043d6f62008-01-07 17:19:16 +0000904functionality with a subclass. Here is how to add a calculated field and
Christian Heimesfe337bf2008-03-23 21:54:12 +0000905a fixed-width print format:
Guido van Rossum3d392eb2007-11-16 00:35:22 +0000906
Christian Heimes043d6f62008-01-07 17:19:16 +0000907 >>> class Point(namedtuple('Point', 'x y')):
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700908 __slots__ = ()
909 @property
910 def hypot(self):
911 return (self.x ** 2 + self.y ** 2) ** 0.5
912 def __str__(self):
913 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 +0000914
Georg Brandl0df79792008-10-04 18:33:26 +0000915 >>> for p in Point(3, 4), Point(14, 5/7):
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700916 print(p)
Christian Heimes25bb7832008-01-11 16:17:00 +0000917 Point: x= 3.000 y= 4.000 hypot= 5.000
918 Point: x=14.000 y= 0.714 hypot=14.018
Christian Heimes043d6f62008-01-07 17:19:16 +0000919
Georg Brandlaf5c2382009-12-28 08:02:38 +0000920The subclass shown above sets ``__slots__`` to an empty tuple. This helps
Christian Heimes679db4a2008-01-18 09:56:22 +0000921keep memory requirements low by preventing the creation of instance dictionaries.
922
Christian Heimes2380ac72008-01-09 00:17:24 +0000923Subclassing is not useful for adding new, stored fields. Instead, simply
Christian Heimesfe337bf2008-03-23 21:54:12 +0000924create a new named tuple type from the :attr:`_fields` attribute:
Christian Heimes2380ac72008-01-09 00:17:24 +0000925
Christian Heimes25bb7832008-01-11 16:17:00 +0000926 >>> Point3D = namedtuple('Point3D', Point._fields + ('z',))
Christian Heimes2380ac72008-01-09 00:17:24 +0000927
Raymond Hettingereac503a2015-05-13 01:09:59 -0700928Docstrings can be customized by making direct assignments to the ``__doc__``
929fields:
930
931 >>> Book = namedtuple('Book', ['id', 'title', 'authors'])
932 >>> Book.__doc__ = 'Hardcover book in active collection'
Berker Peksagde7cafa2015-05-13 12:16:27 +0300933 >>> Book.id.__doc__ = '13-digit ISBN'
934 >>> Book.title.__doc__ = 'Title of first printing'
935 >>> Book.author.__doc__ = 'List of authors sorted by last name'
Raymond Hettingereac503a2015-05-13 01:09:59 -0700936
Christian Heimes2380ac72008-01-09 00:17:24 +0000937Default values can be implemented by using :meth:`_replace` to
Christian Heimesfe337bf2008-03-23 21:54:12 +0000938customize a prototype instance:
Guido van Rossum3d392eb2007-11-16 00:35:22 +0000939
940 >>> Account = namedtuple('Account', 'owner balance transaction_count')
Christian Heimes587c2bf2008-01-19 16:21:02 +0000941 >>> default_account = Account('<owner name>', 0.0, 0)
942 >>> johns_account = default_account._replace(owner='John')
Raymond Hettingerb2d09452011-03-22 22:36:21 -0700943 >>> janes_account = default_account._replace(owner='Jane')
Guido van Rossum3d392eb2007-11-16 00:35:22 +0000944
Christian Heimese4ca8152008-05-08 17:18:53 +0000945Enumerated constants can be implemented with named tuples, but it is simpler
Georg Brandl8ed75cd2014-10-31 10:25:48 +0100946and more efficient to use a simple :class:`~enum.Enum`:
Christian Heimese4ca8152008-05-08 17:18:53 +0000947
948 >>> Status = namedtuple('Status', 'open pending closed')._make(range(3))
949 >>> Status.open, Status.pending, Status.closed
950 (0, 1, 2)
Ezio Melotti993c7822014-10-28 15:07:14 +0100951 >>> from enum import Enum
952 >>> class Status(Enum):
953 ... open, pending, closed = range(3)
Christian Heimese4ca8152008-05-08 17:18:53 +0000954
Georg Brandl8ed75cd2014-10-31 10:25:48 +0100955
956.. seealso::
957
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700958 * `Recipe for named tuple abstract base class with a metaclass mix-in
Raymond Hettingerbfcb4292012-06-10 11:39:44 -0700959 <http://code.activestate.com/recipes/577629-namedtupleabc-abstract-base-class-mix-in-for-named/>`_
960 by Jan Kaliszewski. Besides providing an :term:`abstract base class` for
961 named tuples, it also supports an alternate :term:`metaclass`-based
962 constructor that is convenient for use cases where named tuples are being
963 subclassed.
Raymond Hettingere4c96ad2008-02-06 01:23:58 +0000964
965
Raymond Hettinger2d32f632009-03-02 21:24:57 +0000966:class:`OrderedDict` objects
967----------------------------
968
969Ordered dictionaries are just like regular dictionaries but they remember the
970order that items were inserted. When iterating over an ordered dictionary,
971the items are returned in the order their keys were first added.
972
973.. class:: OrderedDict([items])
974
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700975 Return an instance of a dict subclass, supporting the usual :class:`dict`
976 methods. An *OrderedDict* is a dict that remembers the order that keys
977 were first inserted. If a new entry overwrites an existing entry, the
978 original insertion position is left unchanged. Deleting an entry and
979 reinserting it will move it to the end.
Raymond Hettinger2d32f632009-03-02 21:24:57 +0000980
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700981 .. versionadded:: 3.1
Raymond Hettinger2d32f632009-03-02 21:24:57 +0000982
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700983 .. method:: popitem(last=True)
Raymond Hettingerdc879f02009-03-19 20:30:56 +0000984
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700985 The :meth:`popitem` method for ordered dictionaries returns and removes a
986 (key, value) pair. The pairs are returned in LIFO order if *last* is true
987 or FIFO order if false.
Raymond Hettinger2d32f632009-03-02 21:24:57 +0000988
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700989 .. method:: move_to_end(key, last=True)
Raymond Hettingerf45abc92010-09-06 21:26:09 +0000990
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700991 Move an existing *key* to either end of an ordered dictionary. The item
992 is moved to the right end if *last* is true (the default) or to the
993 beginning if *last* is false. Raises :exc:`KeyError` if the *key* does
994 not exist::
Raymond Hettingerf45abc92010-09-06 21:26:09 +0000995
Raymond Hettinger7929cfb2012-06-09 19:15:26 -0700996 >>> d = OrderedDict.fromkeys('abcde')
997 >>> d.move_to_end('b')
998 >>> ''.join(d.keys())
999 'acdeb'
1000 >>> d.move_to_end('b', last=False)
1001 >>> ''.join(d.keys())
1002 'bacde'
Raymond Hettingerf45abc92010-09-06 21:26:09 +00001003
Raymond Hettinger7929cfb2012-06-09 19:15:26 -07001004 .. versionadded:: 3.2
Raymond Hettingerf45abc92010-09-06 21:26:09 +00001005
Raymond Hettingere9091502009-05-19 17:40:07 +00001006In addition to the usual mapping methods, ordered dictionaries also support
1007reverse iteration using :func:`reversed`.
1008
Raymond Hettinger2d32f632009-03-02 21:24:57 +00001009Equality tests between :class:`OrderedDict` objects are order-sensitive
1010and are implemented as ``list(od1.items())==list(od2.items())``.
1011Equality tests between :class:`OrderedDict` objects and other
Serhiy Storchakabfdcd432013-10-13 23:09:14 +03001012:class:`~collections.abc.Mapping` objects are order-insensitive like regular
1013dictionaries. This allows :class:`OrderedDict` objects to be substituted
1014anywhere a regular dictionary is used.
Raymond Hettinger2d32f632009-03-02 21:24:57 +00001015
Raymond Hettinger36180782009-04-09 22:34:23 +00001016The :class:`OrderedDict` constructor and :meth:`update` method both accept
1017keyword arguments, but their order is lost because Python's function call
1018semantics pass-in keyword arguments using a regular unordered dictionary.
1019
Serhiy Storchaka578c9212014-04-04 15:19:36 +03001020.. versionchanged:: 3.5
1021 The items, keys, and values :term:`views <view>` of :class:`OrderedDict` now
1022 support reverse iteration using :func:`reversed`.
Raymond Hettingerdc879f02009-03-19 20:30:56 +00001023
Raymond Hettinger7bba6832011-04-15 17:43:19 -07001024:class:`OrderedDict` Examples and Recipes
1025^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
1026
Raymond Hettinger0e312012009-11-10 18:35:46 +00001027Since an ordered dictionary remembers its insertion order, it can be used
Donald Stufft8b852f12014-05-20 12:58:38 -04001028in conjunction with sorting to make a sorted dictionary::
Raymond Hettinger0e312012009-11-10 18:35:46 +00001029
1030 >>> # regular unsorted dictionary
1031 >>> d = {'banana': 3, 'apple':4, 'pear': 1, 'orange': 2}
1032
1033 >>> # dictionary sorted by key
1034 >>> OrderedDict(sorted(d.items(), key=lambda t: t[0]))
1035 OrderedDict([('apple', 4), ('banana', 3), ('orange', 2), ('pear', 1)])
1036
1037 >>> # dictionary sorted by value
1038 >>> OrderedDict(sorted(d.items(), key=lambda t: t[1]))
1039 OrderedDict([('pear', 1), ('orange', 2), ('banana', 3), ('apple', 4)])
1040
1041 >>> # dictionary sorted by length of the key string
1042 >>> OrderedDict(sorted(d.items(), key=lambda t: len(t[0])))
1043 OrderedDict([('pear', 1), ('apple', 4), ('orange', 2), ('banana', 3)])
1044
1045The new sorted dictionaries maintain their sort order when entries
1046are deleted. But when new keys are added, the keys are appended
1047to the end and the sort is not maintained.
1048
Raymond Hettinger4821ef82010-07-31 10:14:41 +00001049It is also straight-forward to create an ordered dictionary variant
Andrew Svetlovff63e7a2012-08-31 13:54:54 +03001050that remembers the order the keys were *last* inserted.
Raymond Hettinger4821ef82010-07-31 10:14:41 +00001051If a new entry overwrites an existing entry, the
1052original insertion position is changed and moved to the end::
1053
1054 class LastUpdatedOrderedDict(OrderedDict):
Georg Brandl77570e22010-12-18 16:21:58 +00001055 'Store items in the order the keys were last added'
Raymond Hettinger7bba6832011-04-15 17:43:19 -07001056
Raymond Hettinger4821ef82010-07-31 10:14:41 +00001057 def __setitem__(self, key, value):
1058 if key in self:
1059 del self[key]
1060 OrderedDict.__setitem__(self, key, value)
1061
Éric Araujo889a7dc2011-08-19 00:40:46 +02001062An ordered dictionary can be combined with the :class:`Counter` class
Raymond Hettinger7bba6832011-04-15 17:43:19 -07001063so that the counter remembers the order elements are first encountered::
1064
Raymond Hettinger7929cfb2012-06-09 19:15:26 -07001065 class OrderedCounter(Counter, OrderedDict):
Raymond Hettinger7bba6832011-04-15 17:43:19 -07001066 'Counter that remembers the order elements are first encountered'
1067
Raymond Hettinger7bba6832011-04-15 17:43:19 -07001068 def __repr__(self):
1069 return '%s(%r)' % (self.__class__.__name__, OrderedDict(self))
1070
1071 def __reduce__(self):
1072 return self.__class__, (OrderedDict(self),)
1073
Raymond Hettingere4c96ad2008-02-06 01:23:58 +00001074
1075:class:`UserDict` objects
Mark Summerfield8f2d0062008-02-06 13:30:44 +00001076-------------------------
Raymond Hettingere4c96ad2008-02-06 01:23:58 +00001077
Georg Brandl48310cd2009-01-03 21:18:54 +00001078The class, :class:`UserDict` acts as a wrapper around dictionary objects.
1079The need for this class has been partially supplanted by the ability to
Raymond Hettingere4c96ad2008-02-06 01:23:58 +00001080subclass directly from :class:`dict`; however, this class can be easier
1081to work with because the underlying dictionary is accessible as an
1082attribute.
1083
1084.. class:: UserDict([initialdata])
1085
Raymond Hettinger7929cfb2012-06-09 19:15:26 -07001086 Class that simulates a dictionary. The instance's contents are kept in a
1087 regular dictionary, which is accessible via the :attr:`data` attribute of
1088 :class:`UserDict` instances. If *initialdata* is provided, :attr:`data` is
1089 initialized with its contents; note that a reference to *initialdata* will not
1090 be kept, allowing it be used for other purposes.
Raymond Hettingere4c96ad2008-02-06 01:23:58 +00001091
Raymond Hettinger7929cfb2012-06-09 19:15:26 -07001092 In addition to supporting the methods and operations of mappings,
1093 :class:`UserDict` instances provide the following attribute:
Raymond Hettingere4c96ad2008-02-06 01:23:58 +00001094
Raymond Hettinger7929cfb2012-06-09 19:15:26 -07001095 .. attribute:: data
Raymond Hettingere4c96ad2008-02-06 01:23:58 +00001096
Raymond Hettinger7929cfb2012-06-09 19:15:26 -07001097 A real dictionary used to store the contents of the :class:`UserDict`
1098 class.
Raymond Hettinger53dbe392008-02-12 20:03:09 +00001099
1100
1101
1102:class:`UserList` objects
1103-------------------------
1104
1105This class acts as a wrapper around list objects. It is a useful base class
Georg Brandl48310cd2009-01-03 21:18:54 +00001106for your own list-like classes which can inherit from them and override
Raymond Hettinger53dbe392008-02-12 20:03:09 +00001107existing methods or add new ones. In this way, one can add new behaviors to
1108lists.
1109
Georg Brandl48310cd2009-01-03 21:18:54 +00001110The need for this class has been partially supplanted by the ability to
Raymond Hettinger53dbe392008-02-12 20:03:09 +00001111subclass directly from :class:`list`; however, this class can be easier
1112to work with because the underlying list is accessible as an attribute.
1113
1114.. class:: UserList([list])
1115
Raymond Hettinger7929cfb2012-06-09 19:15:26 -07001116 Class that simulates a list. The instance's contents are kept in a regular
1117 list, which is accessible via the :attr:`data` attribute of :class:`UserList`
1118 instances. The instance's contents are initially set to a copy of *list*,
1119 defaulting to the empty list ``[]``. *list* can be any iterable, for
1120 example a real Python list or a :class:`UserList` object.
Raymond Hettinger53dbe392008-02-12 20:03:09 +00001121
Raymond Hettinger7929cfb2012-06-09 19:15:26 -07001122 In addition to supporting the methods and operations of mutable sequences,
1123 :class:`UserList` instances provide the following attribute:
Raymond Hettinger53dbe392008-02-12 20:03:09 +00001124
Raymond Hettinger7929cfb2012-06-09 19:15:26 -07001125 .. attribute:: data
Raymond Hettinger53dbe392008-02-12 20:03:09 +00001126
Raymond Hettinger7929cfb2012-06-09 19:15:26 -07001127 A real :class:`list` object used to store the contents of the
1128 :class:`UserList` class.
Raymond Hettinger53dbe392008-02-12 20:03:09 +00001129
Zachary Ware80602e02014-01-13 20:38:57 -06001130**Subclassing requirements:** Subclasses of :class:`UserList` are expected to
Raymond Hettinger53dbe392008-02-12 20:03:09 +00001131offer a constructor which can be called with either no arguments or one
1132argument. List operations which return a new sequence attempt to create an
1133instance of the actual implementation class. To do so, it assumes that the
1134constructor can be called with a single parameter, which is a sequence object
1135used as a data source.
1136
1137If a derived class does not wish to comply with this requirement, all of the
1138special methods supported by this class will need to be overridden; please
1139consult the sources for information about the methods which need to be provided
1140in that case.
Raymond Hettingerb3a65f82008-02-21 22:11:37 +00001141
1142:class:`UserString` objects
Christian Heimesc3f30c42008-02-22 16:37:40 +00001143---------------------------
Raymond Hettingerb3a65f82008-02-21 22:11:37 +00001144
Georg Brandl48310cd2009-01-03 21:18:54 +00001145The class, :class:`UserString` acts as a wrapper around string objects.
1146The need for this class has been partially supplanted by the ability to
Raymond Hettingerb3a65f82008-02-21 22:11:37 +00001147subclass directly from :class:`str`; however, this class can be easier
1148to work with because the underlying string is accessible as an
1149attribute.
1150
1151.. class:: UserString([sequence])
1152
Raymond Hettinger7929cfb2012-06-09 19:15:26 -07001153 Class that simulates a string or a Unicode string object. The instance's
1154 content is kept in a regular string object, which is accessible via the
1155 :attr:`data` attribute of :class:`UserString` instances. The instance's
1156 contents are initially set to a copy of *sequence*. The *sequence* can
1157 be an instance of :class:`bytes`, :class:`str`, :class:`UserString` (or a
1158 subclass) or an arbitrary sequence which can be converted into a string using
1159 the built-in :func:`str` function.