blob: 188fd0a1118354f27533c637888e7ef9badedcfb [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 Hettinger53dbe392008-02-12 20:03:09 +00005 :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
11 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
Éric Araujof90112e2011-06-03 19:18:41 +020036 Moved :ref:`collections-abstract-base-classes` to the :mod:`collections.abc` module.
Raymond Hettinger158c9c22011-02-22 00:41:50 +000037 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
44A :class:`ChainMap` class is provided for quickly linking a number of mappings
45so they can be treated as a single unit. It is often much faster than creating
46a new dictionary and running multiple :meth:`~dict.update` calls.
47
48The class can be used to simulate nested scopes and is useful in templating.
49
50.. class:: ChainMap(*maps)
51
52 A :class:`ChainMap` groups multiple dicts or other mappings together to
53 create a single, updateable view. If no *maps* are specified, a single empty
54 dictionary is provided so that a new chain always has at least one mapping.
55
56 The underlying mappings are stored in a list. That list is public and can
57 accessed or updated using the *maps* attribute. There is no other state.
58
59 Lookups search the underlying mappings successively until a key is found. In
60 contrast, writes, updates, and deletions only operate on the first mapping.
61
Raymond Hettinger4bbde722011-04-11 17:57:21 -070062 A :class:`ChainMap` incorporates the underlying mappings by reference. So, if
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +000063 one of the underlying mappings gets updated, those changes will be reflected
Raymond Hettinger4bbde722011-04-11 17:57:21 -070064 in :class:`ChainMap`.
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +000065
66 All of the usual dictionary methods are supported. In addition, there is a
67 *maps* attribute, a method for creating new subcontexts, and a property for
68 accessing all but the first mapping:
69
70 .. attribute:: maps
71
72 A user updateable list of mappings. The list is ordered from
73 first-searched to last-searched. It is the only stored state and can
Éric Araujoe5905a92011-08-16 19:09:56 +020074 be modified to change which mappings are searched. The list should
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +000075 always contain at least one mapping.
76
77 .. method:: new_child()
78
79 Returns a new :class:`ChainMap` containing a new :class:`dict` followed by
80 all of the maps in the current instance. A call to ``d.new_child()`` is
81 equivalent to: ``ChainMap({}, *d.maps)``. This method is used for
82 creating subcontexts that can be updated without altering values in any
83 of the parent mappings.
84
Éric Araujo7ebadd52011-06-11 03:27:03 +020085 .. method:: parents()
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +000086
87 Returns a new :class:`ChainMap` containing all of the maps in the current
88 instance except the first one. This is useful for skipping the first map
89 in the search. The use-cases are similar to those for the
90 :keyword:`nonlocal` keyword used in :term:`nested scopes <nested scope>`.
91 The use-cases also parallel those for the builtin :func:`super` function.
92 A reference to ``d.parents`` is equivalent to: ``ChainMap(*d.maps[1:])``.
93
94 .. versionadded:: 3.3
95
96 Example of simulating Python's internal lookup chain::
97
Raymond Hettinger94ca2112011-02-26 02:48:44 +000098 import builtins
99 pylookup = ChainMap(locals(), globals(), vars(builtins))
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +0000100
101 Example of letting user specified values take precedence over environment
102 variables which in turn take precedence over default values::
103
104 import os, argparse
105 defaults = {'color': 'red', 'user': guest}
106 parser = argparse.ArgumentParser()
107 parser.add_argument('-u', '--user')
108 parser.add_argument('-c', '--color')
109 user_specified = vars(parser.parse_args())
110 combined = ChainMap(user_specified, os.environ, defaults)
111
112 Example patterns for using the :class:`ChainMap` class to simulate nested
113 contexts::
114
115 c = ChainMap() Create root context
116 d = c.new_child() Create nested child context
117 e = c.new_child() Child of c, independent from d
118 e.maps[0] Current context dictionary -- like Python's locals()
119 e.maps[-1] Root context -- like Python's globals()
120 e.parents Enclosing context chain -- like Python's nonlocals
121
122 d['x'] Get first key in the chain of contexts
123 d['x'] = 1 Set value in current context
124 del['x'] Delete from current context
125 list(d) All nested values
126 k in d Check all nested values
127 len(d) Number of nested values
128 d.items() All nested items
129 dict(d) Flatten into a regular dictionary
130
131 .. seealso::
132
133 * The `MultiContext class
134 <http://svn.enthought.com/svn/enthought/CodeTools/trunk/enthought/contexts/multi_context.py>`_
135 in the Enthought `CodeTools package
Éric Araujo405d7782011-04-24 03:00:58 +0200136 <https://github.com/enthought/codetools>`_ has options to support
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +0000137 writing to any mapping in the chain.
138
139 * Django's `Context class
140 <http://code.djangoproject.com/browser/django/trunk/django/template/context.py>`_
141 for templating is a read-only chain of mappings. It also features
142 pushing and popping of contexts similar to the
143 :meth:`~collections.ChainMap.new_child` method and the
144 :meth:`~collections.ChainMap.parents` property.
145
146 * The `Nested Contexts recipe
147 <http://code.activestate.com/recipes/577434/>`_ has options to control
148 whether writes and other mutations apply only to the first mapping or to
149 any mapping in the chain.
150
151 * A `greatly simplified read-only version of Chainmap
Éric Araujo405d7782011-04-24 03:00:58 +0200152 <http://code.activestate.com/recipes/305268/>`_.
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +0000153
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000154:class:`Counter` objects
155------------------------
156
157A counter tool is provided to support convenient and rapid tallies.
158For example::
159
Raymond Hettinger1c62dc92009-02-04 11:41:45 +0000160 >>> # Tally occurrences of words in a list
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000161 >>> cnt = Counter()
Raymond Hettinger670eaec2009-01-21 23:14:07 +0000162 >>> for word in ['red', 'blue', 'red', 'green', 'blue', 'blue']:
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000163 ... cnt[word] += 1
164 >>> cnt
165 Counter({'blue': 3, 'red': 2, 'green': 1})
166
Raymond Hettinger1c62dc92009-02-04 11:41:45 +0000167 >>> # Find the ten most common words in Hamlet
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000168 >>> import re
169 >>> words = re.findall('\w+', open('hamlet.txt').read().lower())
Raymond Hettinger0bae6622009-01-20 13:00:59 +0000170 >>> Counter(words).most_common(10)
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000171 [('the', 1143), ('and', 966), ('to', 762), ('of', 669), ('i', 631),
172 ('you', 554), ('a', 546), ('my', 514), ('hamlet', 471), ('in', 451)]
173
174.. class:: Counter([iterable-or-mapping])
175
Raymond Hettinger670eaec2009-01-21 23:14:07 +0000176 A :class:`Counter` is a :class:`dict` subclass for counting hashable objects.
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000177 It is an unordered collection where elements are stored as dictionary keys
178 and their counts are stored as dictionary values. Counts are allowed to be
179 any integer value including zero or negative counts. The :class:`Counter`
180 class is similar to bags or multisets in other languages.
181
182 Elements are counted from an *iterable* or initialized from another
Benjamin Peterson25c95f12009-05-08 20:42:26 +0000183 *mapping* (or counter):
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000184
Raymond Hettinger73662a52009-01-27 02:38:22 +0000185 >>> c = Counter() # a new, empty counter
186 >>> c = Counter('gallahad') # a new counter from an iterable
187 >>> c = Counter({'red': 4, 'blue': 2}) # a new counter from a mapping
188 >>> c = Counter(cats=4, dogs=8) # a new counter from keyword args
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000189
Raymond Hettinger670eaec2009-01-21 23:14:07 +0000190 Counter objects have a dictionary interface except that they return a zero
Benjamin Peterson25c95f12009-05-08 20:42:26 +0000191 count for missing items instead of raising a :exc:`KeyError`:
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000192
Raymond Hettinger94adc8e2009-01-22 05:27:37 +0000193 >>> c = Counter(['eggs', 'ham'])
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000194 >>> c['bacon'] # count of a missing element is zero
195 0
196
Raymond Hettinger94adc8e2009-01-22 05:27:37 +0000197 Setting a count to zero does not remove an element from a counter.
198 Use ``del`` to remove it entirely:
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000199
Raymond Hettinger94adc8e2009-01-22 05:27:37 +0000200 >>> c['sausage'] = 0 # counter entry with a zero count
201 >>> del c['sausage'] # del actually removes the entry
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000202
Benjamin Petersond45bf582009-03-02 21:44:54 +0000203 .. versionadded:: 3.1
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000204
205
Ezio Melotti0be8b1c2010-04-04 06:53:44 +0000206 Counter objects support three methods beyond those available for all
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000207 dictionaries:
208
209 .. method:: elements()
210
Raymond Hettinger670eaec2009-01-21 23:14:07 +0000211 Return an iterator over elements repeating each as many times as its
212 count. Elements are returned in arbitrary order. If an element's count
213 is less than one, :meth:`elements` will ignore it.
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000214
Raymond Hettinger0bae6622009-01-20 13:00:59 +0000215 >>> c = Counter(a=4, b=2, c=0, d=-2)
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000216 >>> list(c.elements())
217 ['a', 'a', 'a', 'a', 'b', 'b']
218
219 .. method:: most_common([n])
220
Raymond Hettinger73662a52009-01-27 02:38:22 +0000221 Return a list of the *n* most common elements and their counts from the
Raymond Hettingerd04fa312009-02-04 19:45:13 +0000222 most common to the least. If *n* is not specified, :func:`most_common`
Raymond Hettinger73662a52009-01-27 02:38:22 +0000223 returns *all* elements in the counter. Elements with equal counts are
Benjamin Peterson25c95f12009-05-08 20:42:26 +0000224 ordered arbitrarily:
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000225
226 >>> Counter('abracadabra').most_common(3)
227 [('a', 5), ('r', 2), ('b', 2)]
228
Raymond Hettinger9c01e442010-04-03 10:32:58 +0000229 .. method:: subtract([iterable-or-mapping])
230
231 Elements are subtracted from an *iterable* or from another *mapping*
232 (or counter). Like :meth:`dict.update` but subtracts counts instead
233 of replacing them. Both inputs and outputs may be zero or negative.
234
235 >>> c = Counter(a=4, b=2, c=0, d=-2)
236 >>> d = Counter(a=1, b=2, c=3, d=4)
237 >>> c.subtract(d)
238 Counter({'a': 3, 'b': 0, 'c': -3, 'd': -6})
239
Ezio Melotti0be8b1c2010-04-04 06:53:44 +0000240 .. versionadded:: 3.2
241
Raymond Hettinger670eaec2009-01-21 23:14:07 +0000242 The usual dictionary methods are available for :class:`Counter` objects
243 except for two which work differently for counters.
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000244
245 .. method:: fromkeys(iterable)
246
Raymond Hettinger73662a52009-01-27 02:38:22 +0000247 This class method is not implemented for :class:`Counter` objects.
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000248
249 .. method:: update([iterable-or-mapping])
250
Raymond Hettinger73662a52009-01-27 02:38:22 +0000251 Elements are counted from an *iterable* or added-in from another
252 *mapping* (or counter). Like :meth:`dict.update` but adds counts
253 instead of replacing them. Also, the *iterable* is expected to be a
254 sequence of elements, not a sequence of ``(key, value)`` pairs.
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000255
256Common patterns for working with :class:`Counter` objects::
257
Raymond Hettinger73662a52009-01-27 02:38:22 +0000258 sum(c.values()) # total of all counts
259 c.clear() # reset all counts
260 list(c) # list unique elements
261 set(c) # convert to a set
262 dict(c) # convert to a regular dictionary
263 c.items() # convert to a list of (elem, cnt) pairs
264 Counter(dict(list_of_pairs)) # convert from a list of (elem, cnt) pairs
265 c.most_common()[:-n:-1] # n least common elements
Raymond Hettingerfcb393c2011-08-09 13:00:40 -0700266 +c # remove zero and negative counts
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000267
Raymond Hettinger72a95cc2009-02-25 22:51:40 +0000268Several mathematical operations are provided for combining :class:`Counter`
269objects to produce multisets (counters that have counts greater than zero).
270Addition and subtraction combine counters by adding or subtracting the counts
271of corresponding elements. Intersection and union return the minimum and
272maximum of corresponding counts. Each operation can accept inputs with signed
273counts, but the output will exclude results with counts of zero or less.
Raymond Hettinger4d2073a2009-01-20 03:41:22 +0000274
Raymond Hettingere0d1b9f2009-01-21 20:36:27 +0000275 >>> c = Counter(a=3, b=1)
276 >>> d = Counter(a=1, b=2)
Raymond Hettinger73662a52009-01-27 02:38:22 +0000277 >>> c + d # add two counters together: c[x] + d[x]
Raymond Hettinger4d2073a2009-01-20 03:41:22 +0000278 Counter({'a': 4, 'b': 3})
Raymond Hettinger73662a52009-01-27 02:38:22 +0000279 >>> c - d # subtract (keeping only positive counts)
Raymond Hettinger4d2073a2009-01-20 03:41:22 +0000280 Counter({'a': 2})
Raymond Hettinger73662a52009-01-27 02:38:22 +0000281 >>> c & d # intersection: min(c[x], d[x])
Raymond Hettinger4d2073a2009-01-20 03:41:22 +0000282 Counter({'a': 1, 'b': 1})
Raymond Hettinger73662a52009-01-27 02:38:22 +0000283 >>> c | d # union: max(c[x], d[x])
Raymond Hettinger4d2073a2009-01-20 03:41:22 +0000284 Counter({'a': 3, 'b': 2})
285
Raymond Hettingerfcb393c2011-08-09 13:00:40 -0700286Unary addition and substraction are shortcuts for adding an empty counter
287or subtracting from an empty counter.
288
289 >>> c = Counter(a=2, b=-4)
290 >>> +c
291 Counter({'a': 2})
292 >>> -c
293 Counter({'b': 4})
294
295.. versionadded:: 3.3
Raymond Hettingerbecd5682011-10-19 13:40:37 -0700296 Added support for unary plus, unary minus, and in-place multiset operations.
Raymond Hettingerfcb393c2011-08-09 13:00:40 -0700297
Raymond Hettinger22f18852010-04-12 21:45:14 +0000298.. note::
299
300 Counters were primarily designed to work with positive integers to represent
301 running counts; however, care was taken to not unnecessarily preclude use
302 cases needing other types or negative values. To help with those use cases,
303 this section documents the minimum range and type restrictions.
304
305 * The :class:`Counter` class itself is a dictionary subclass with no
306 restrictions on its keys and values. The values are intended to be numbers
307 representing counts, but you *could* store anything in the value field.
308
309 * The :meth:`most_common` method requires only that the values be orderable.
310
311 * For in-place operations such as ``c[key] += 1``, the value type need only
312 support addition and subtraction. So fractions, floats, and decimals would
313 work and negative values are supported. The same is also true for
314 :meth:`update` and :meth:`subtract` which allow negative and zero values
315 for both inputs and outputs.
316
317 * The multiset methods are designed only for use cases with positive values.
318 The inputs may be negative or zero, but only outputs with positive values
319 are created. There are no type restrictions, but the value type needs to
Ezio Melottie130a522011-10-19 10:58:56 +0300320 support addition, subtraction, and comparison.
Raymond Hettinger22f18852010-04-12 21:45:14 +0000321
322 * The :meth:`elements` method requires integer counts. It ignores zero and
323 negative counts.
324
Raymond Hettingerb14043c2009-01-20 23:44:31 +0000325.. seealso::
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000326
Raymond Hettinger94adc8e2009-01-22 05:27:37 +0000327 * `Counter class <http://code.activestate.com/recipes/576611/>`_
328 adapted for Python 2.5 and an early `Bag recipe
329 <http://code.activestate.com/recipes/259174/>`_ for Python 2.4.
330
Raymond Hettingerb14043c2009-01-20 23:44:31 +0000331 * `Bag class <http://www.gnu.org/software/smalltalk/manual-base/html_node/Bag.html>`_
332 in Smalltalk.
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000333
Éric Araujo08c9bd52011-04-24 02:59:02 +0200334 * Wikipedia entry for `Multisets <http://en.wikipedia.org/wiki/Multiset>`_.
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000335
Raymond Hettingerb14043c2009-01-20 23:44:31 +0000336 * `C++ multisets <http://www.demo2s.com/Tutorial/Cpp/0380__set-multiset/Catalog0380__set-multiset.htm>`_
Raymond Hettinger94adc8e2009-01-22 05:27:37 +0000337 tutorial with examples.
Raymond Hettingerb14043c2009-01-20 23:44:31 +0000338
Raymond Hettinger94adc8e2009-01-22 05:27:37 +0000339 * For mathematical operations on multisets and their use cases, see
Raymond Hettingerb14043c2009-01-20 23:44:31 +0000340 *Knuth, Donald. The Art of Computer Programming Volume II,
Éric Araujo08c9bd52011-04-24 02:59:02 +0200341 Section 4.6.3, Exercise 19*.
Raymond Hettingerb14043c2009-01-20 23:44:31 +0000342
Raymond Hettinger670eaec2009-01-21 23:14:07 +0000343 * To enumerate all distinct multisets of a given size over a given set of
Raymond Hettingerd07d9392009-01-27 04:20:44 +0000344 elements, see :func:`itertools.combinations_with_replacement`.
Raymond Hettingerb14043c2009-01-20 23:44:31 +0000345
Raymond Hettinger94adc8e2009-01-22 05:27:37 +0000346 map(Counter, combinations_with_replacement('ABC', 2)) --> AA AB AC BB BC CC
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000347
348
Georg Brandl116aa622007-08-15 14:28:22 +0000349:class:`deque` objects
350----------------------
351
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000352.. class:: deque([iterable, [maxlen]])
Georg Brandl116aa622007-08-15 14:28:22 +0000353
354 Returns a new deque object initialized left-to-right (using :meth:`append`) with
355 data from *iterable*. If *iterable* is not specified, the new deque is empty.
356
357 Deques are a generalization of stacks and queues (the name is pronounced "deck"
358 and is short for "double-ended queue"). Deques support thread-safe, memory
359 efficient appends and pops from either side of the deque with approximately the
360 same O(1) performance in either direction.
361
362 Though :class:`list` objects support similar operations, they are optimized for
363 fast fixed-length operations and incur O(n) memory movement costs for
364 ``pop(0)`` and ``insert(0, v)`` operations which change both the size and
365 position of the underlying data representation.
366
Georg Brandl116aa622007-08-15 14:28:22 +0000367
Georg Brandl9afde1c2007-11-01 20:32:30 +0000368 If *maxlen* is not specified or is *None*, deques may grow to an
369 arbitrary length. Otherwise, the deque is bounded to the specified maximum
370 length. Once a bounded length deque is full, when new items are added, a
371 corresponding number of items are discarded from the opposite end. Bounded
372 length deques provide functionality similar to the ``tail`` filter in
373 Unix. They are also useful for tracking transactions and other pools of data
374 where only the most recent activity is of interest.
375
Georg Brandl9afde1c2007-11-01 20:32:30 +0000376
Benjamin Petersone41251e2008-04-25 01:59:09 +0000377 Deque objects support the following methods:
Georg Brandl116aa622007-08-15 14:28:22 +0000378
Benjamin Petersone41251e2008-04-25 01:59:09 +0000379 .. method:: append(x)
Georg Brandl116aa622007-08-15 14:28:22 +0000380
Benjamin Petersone41251e2008-04-25 01:59:09 +0000381 Add *x* to the right side of the deque.
Georg Brandl116aa622007-08-15 14:28:22 +0000382
383
Benjamin Petersone41251e2008-04-25 01:59:09 +0000384 .. method:: appendleft(x)
Georg Brandl116aa622007-08-15 14:28:22 +0000385
Benjamin Petersone41251e2008-04-25 01:59:09 +0000386 Add *x* to the left side of the deque.
Georg Brandl116aa622007-08-15 14:28:22 +0000387
388
Benjamin Petersone41251e2008-04-25 01:59:09 +0000389 .. method:: clear()
Georg Brandl116aa622007-08-15 14:28:22 +0000390
Benjamin Petersone41251e2008-04-25 01:59:09 +0000391 Remove all elements from the deque leaving it with length 0.
Georg Brandl116aa622007-08-15 14:28:22 +0000392
393
Raymond Hettinger44459de2010-04-03 23:20:46 +0000394 .. method:: count(x)
395
396 Count the number of deque elements equal to *x*.
397
398 .. versionadded:: 3.2
399
Georg Brandl67b21b72010-08-17 15:07:14 +0000400
Benjamin Petersone41251e2008-04-25 01:59:09 +0000401 .. method:: extend(iterable)
Georg Brandl116aa622007-08-15 14:28:22 +0000402
Benjamin Petersone41251e2008-04-25 01:59:09 +0000403 Extend the right side of the deque by appending elements from the iterable
404 argument.
Georg Brandl116aa622007-08-15 14:28:22 +0000405
406
Benjamin Petersone41251e2008-04-25 01:59:09 +0000407 .. method:: extendleft(iterable)
Georg Brandl116aa622007-08-15 14:28:22 +0000408
Benjamin Petersone41251e2008-04-25 01:59:09 +0000409 Extend the left side of the deque by appending elements from *iterable*.
410 Note, the series of left appends results in reversing the order of
411 elements in the iterable argument.
Georg Brandl116aa622007-08-15 14:28:22 +0000412
413
Benjamin Petersone41251e2008-04-25 01:59:09 +0000414 .. method:: pop()
Georg Brandl116aa622007-08-15 14:28:22 +0000415
Benjamin Petersone41251e2008-04-25 01:59:09 +0000416 Remove and return an element from the right side of the deque. If no
417 elements are present, raises an :exc:`IndexError`.
Georg Brandl116aa622007-08-15 14:28:22 +0000418
419
Benjamin Petersone41251e2008-04-25 01:59:09 +0000420 .. method:: popleft()
Georg Brandl116aa622007-08-15 14:28:22 +0000421
Benjamin Petersone41251e2008-04-25 01:59:09 +0000422 Remove and return an element from the left side of the deque. If no
423 elements are present, raises an :exc:`IndexError`.
Georg Brandl116aa622007-08-15 14:28:22 +0000424
425
Benjamin Petersone41251e2008-04-25 01:59:09 +0000426 .. method:: remove(value)
Georg Brandl116aa622007-08-15 14:28:22 +0000427
Benjamin Petersone41251e2008-04-25 01:59:09 +0000428 Removed the first occurrence of *value*. If not found, raises a
429 :exc:`ValueError`.
Georg Brandl116aa622007-08-15 14:28:22 +0000430
Georg Brandl67b21b72010-08-17 15:07:14 +0000431
Raymond Hettingere5fdedb2009-12-10 00:47:21 +0000432 .. method:: reverse()
433
434 Reverse the elements of the deque in-place and then return ``None``.
435
436 .. versionadded:: 3.2
Georg Brandl116aa622007-08-15 14:28:22 +0000437
Georg Brandl67b21b72010-08-17 15:07:14 +0000438
Benjamin Petersone41251e2008-04-25 01:59:09 +0000439 .. method:: rotate(n)
Georg Brandl116aa622007-08-15 14:28:22 +0000440
Benjamin Petersone41251e2008-04-25 01:59:09 +0000441 Rotate the deque *n* steps to the right. If *n* is negative, rotate to
442 the left. Rotating one step to the right is equivalent to:
443 ``d.appendleft(d.pop())``.
444
Georg Brandl116aa622007-08-15 14:28:22 +0000445
Raymond Hettinger5bb0f0e2009-03-10 12:56:32 +0000446 Deque objects also provide one read-only attribute:
447
448 .. attribute:: maxlen
449
450 Maximum size of a deque or *None* if unbounded.
451
Raymond Hettinger150fb9c2009-03-10 22:48:06 +0000452 .. versionadded:: 3.1
Raymond Hettinger5bb0f0e2009-03-10 12:56:32 +0000453
454
Georg Brandl116aa622007-08-15 14:28:22 +0000455In addition to the above, deques support iteration, pickling, ``len(d)``,
456``reversed(d)``, ``copy.copy(d)``, ``copy.deepcopy(d)``, membership testing with
Benjamin Peterson206e3072008-10-19 14:07:49 +0000457the :keyword:`in` operator, and subscript references such as ``d[-1]``. Indexed
458access is O(1) at both ends but slows to O(n) in the middle. For fast random
459access, use lists instead.
Georg Brandl116aa622007-08-15 14:28:22 +0000460
Christian Heimesfe337bf2008-03-23 21:54:12 +0000461Example:
462
463.. doctest::
Georg Brandl116aa622007-08-15 14:28:22 +0000464
465 >>> from collections import deque
466 >>> d = deque('ghi') # make a new deque with three items
467 >>> for elem in d: # iterate over the deque's elements
Neal Norwitz752abd02008-05-13 04:55:24 +0000468 ... print(elem.upper())
Georg Brandl116aa622007-08-15 14:28:22 +0000469 G
470 H
471 I
472
473 >>> d.append('j') # add a new entry to the right side
474 >>> d.appendleft('f') # add a new entry to the left side
475 >>> d # show the representation of the deque
476 deque(['f', 'g', 'h', 'i', 'j'])
477
478 >>> d.pop() # return and remove the rightmost item
479 'j'
480 >>> d.popleft() # return and remove the leftmost item
481 'f'
482 >>> list(d) # list the contents of the deque
483 ['g', 'h', 'i']
484 >>> d[0] # peek at leftmost item
485 'g'
486 >>> d[-1] # peek at rightmost item
487 'i'
488
489 >>> list(reversed(d)) # list the contents of a deque in reverse
490 ['i', 'h', 'g']
491 >>> 'h' in d # search the deque
492 True
493 >>> d.extend('jkl') # add multiple elements at once
494 >>> d
495 deque(['g', 'h', 'i', 'j', 'k', 'l'])
496 >>> d.rotate(1) # right rotation
497 >>> d
498 deque(['l', 'g', 'h', 'i', 'j', 'k'])
499 >>> d.rotate(-1) # left rotation
500 >>> d
501 deque(['g', 'h', 'i', 'j', 'k', 'l'])
502
503 >>> deque(reversed(d)) # make a new deque in reverse order
504 deque(['l', 'k', 'j', 'i', 'h', 'g'])
505 >>> d.clear() # empty the deque
506 >>> d.pop() # cannot pop from an empty deque
507 Traceback (most recent call last):
508 File "<pyshell#6>", line 1, in -toplevel-
509 d.pop()
510 IndexError: pop from an empty deque
511
512 >>> d.extendleft('abc') # extendleft() reverses the input order
513 >>> d
514 deque(['c', 'b', 'a'])
515
516
Georg Brandl9afde1c2007-11-01 20:32:30 +0000517:class:`deque` Recipes
518^^^^^^^^^^^^^^^^^^^^^^
Georg Brandl116aa622007-08-15 14:28:22 +0000519
520This section shows various approaches to working with deques.
521
Raymond Hettingerd2ee64d2009-03-31 22:52:48 +0000522Bounded length deques provide functionality similar to the ``tail`` filter
523in Unix::
524
525 def tail(filename, n=10):
526 'Return the last n lines of a file'
Éric Araujoa3dd56b2011-03-11 17:42:48 +0100527 with open(filename) as f:
528 return deque(f, n)
Raymond Hettingerd2ee64d2009-03-31 22:52:48 +0000529
530Another approach to using deques is to maintain a sequence of recently
531added elements by appending to the right and popping to the left::
532
533 def moving_average(iterable, n=3):
534 # moving_average([40, 30, 50, 46, 39, 44]) --> 40.0 42.0 45.0 43.0
535 # http://en.wikipedia.org/wiki/Moving_average
536 it = iter(iterable)
Raymond Hettingerd40285a2009-05-22 01:11:26 +0000537 d = deque(itertools.islice(it, n-1))
538 d.appendleft(0)
Raymond Hettingerd2ee64d2009-03-31 22:52:48 +0000539 s = sum(d)
Raymond Hettingerd2ee64d2009-03-31 22:52:48 +0000540 for elem in it:
541 s += elem - d.popleft()
542 d.append(elem)
543 yield s / n
544
Georg Brandl116aa622007-08-15 14:28:22 +0000545The :meth:`rotate` method provides a way to implement :class:`deque` slicing and
Ezio Melotti0639d5a2009-12-19 23:26:38 +0000546deletion. For example, a pure Python implementation of ``del d[n]`` relies on
Georg Brandl116aa622007-08-15 14:28:22 +0000547the :meth:`rotate` method to position elements to be popped::
548
549 def delete_nth(d, n):
550 d.rotate(-n)
551 d.popleft()
552 d.rotate(n)
553
554To implement :class:`deque` slicing, use a similar approach applying
555:meth:`rotate` to bring a target element to the left side of the deque. Remove
556old entries with :meth:`popleft`, add new entries with :meth:`extend`, and then
557reverse the rotation.
Georg Brandl116aa622007-08-15 14:28:22 +0000558With minor variations on that approach, it is easy to implement Forth style
559stack manipulations such as ``dup``, ``drop``, ``swap``, ``over``, ``pick``,
560``rot``, and ``roll``.
561
Georg Brandl116aa622007-08-15 14:28:22 +0000562
563:class:`defaultdict` objects
564----------------------------
565
Georg Brandl116aa622007-08-15 14:28:22 +0000566.. class:: defaultdict([default_factory[, ...]])
567
568 Returns a new dictionary-like object. :class:`defaultdict` is a subclass of the
Georg Brandl22b34312009-07-26 14:54:51 +0000569 built-in :class:`dict` class. It overrides one method and adds one writable
Georg Brandl116aa622007-08-15 14:28:22 +0000570 instance variable. The remaining functionality is the same as for the
571 :class:`dict` class and is not documented here.
572
573 The first argument provides the initial value for the :attr:`default_factory`
574 attribute; it defaults to ``None``. All remaining arguments are treated the same
575 as if they were passed to the :class:`dict` constructor, including keyword
576 arguments.
577
Georg Brandl116aa622007-08-15 14:28:22 +0000578
Benjamin Petersone41251e2008-04-25 01:59:09 +0000579 :class:`defaultdict` objects support the following method in addition to the
580 standard :class:`dict` operations:
Georg Brandl116aa622007-08-15 14:28:22 +0000581
Benjamin Petersond319ad52010-07-18 14:27:02 +0000582 .. method:: __missing__(key)
Georg Brandl116aa622007-08-15 14:28:22 +0000583
Benjamin Peterson5478b472008-09-17 22:25:09 +0000584 If the :attr:`default_factory` attribute is ``None``, this raises a
Benjamin Petersone41251e2008-04-25 01:59:09 +0000585 :exc:`KeyError` exception with the *key* as argument.
Georg Brandl116aa622007-08-15 14:28:22 +0000586
Benjamin Petersone41251e2008-04-25 01:59:09 +0000587 If :attr:`default_factory` is not ``None``, it is called without arguments
588 to provide a default value for the given *key*, this value is inserted in
589 the dictionary for the *key*, and returned.
Georg Brandl116aa622007-08-15 14:28:22 +0000590
Benjamin Petersone41251e2008-04-25 01:59:09 +0000591 If calling :attr:`default_factory` raises an exception this exception is
592 propagated unchanged.
Georg Brandl116aa622007-08-15 14:28:22 +0000593
Benjamin Petersone41251e2008-04-25 01:59:09 +0000594 This method is called by the :meth:`__getitem__` method of the
595 :class:`dict` class when the requested key is not found; whatever it
596 returns or raises is then returned or raised by :meth:`__getitem__`.
Georg Brandl116aa622007-08-15 14:28:22 +0000597
Benjamin Peterson871b9d12012-01-27 09:14:01 -0500598 Note that :meth:`__missing__` is *not* called for any operations besides
599 :meth:`__getitem__`. This means that :meth:`get` will, like normal
600 dictionaries, return ``None`` as a default rather than using
601 :attr:`default_factory`.
602
Georg Brandl116aa622007-08-15 14:28:22 +0000603
Benjamin Petersone41251e2008-04-25 01:59:09 +0000604 :class:`defaultdict` objects support the following instance variable:
Georg Brandl116aa622007-08-15 14:28:22 +0000605
Benjamin Petersone41251e2008-04-25 01:59:09 +0000606
Benjamin Petersond319ad52010-07-18 14:27:02 +0000607 .. attribute:: default_factory
Benjamin Petersone41251e2008-04-25 01:59:09 +0000608
609 This attribute is used by the :meth:`__missing__` method; it is
610 initialized from the first argument to the constructor, if present, or to
611 ``None``, if absent.
Georg Brandl116aa622007-08-15 14:28:22 +0000612
613
Georg Brandl116aa622007-08-15 14:28:22 +0000614:class:`defaultdict` Examples
615^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
616
617Using :class:`list` as the :attr:`default_factory`, it is easy to group a
Christian Heimesfe337bf2008-03-23 21:54:12 +0000618sequence of key-value pairs into a dictionary of lists:
Georg Brandl116aa622007-08-15 14:28:22 +0000619
620 >>> s = [('yellow', 1), ('blue', 2), ('yellow', 3), ('blue', 4), ('red', 1)]
621 >>> d = defaultdict(list)
622 >>> for k, v in s:
623 ... d[k].append(v)
624 ...
Ezio Melottic53a8942009-09-12 01:52:05 +0000625 >>> list(d.items())
Georg Brandl116aa622007-08-15 14:28:22 +0000626 [('blue', [2, 4]), ('red', [1]), ('yellow', [1, 3])]
627
628When each key is encountered for the first time, it is not already in the
629mapping; so an entry is automatically created using the :attr:`default_factory`
630function which returns an empty :class:`list`. The :meth:`list.append`
631operation then attaches the value to the new list. When keys are encountered
632again, the look-up proceeds normally (returning the list for that key) and the
633:meth:`list.append` operation adds another value to the list. This technique is
Christian Heimesfe337bf2008-03-23 21:54:12 +0000634simpler and faster than an equivalent technique using :meth:`dict.setdefault`:
Georg Brandl116aa622007-08-15 14:28:22 +0000635
636 >>> d = {}
637 >>> for k, v in s:
638 ... d.setdefault(k, []).append(v)
639 ...
Ezio Melottic53a8942009-09-12 01:52:05 +0000640 >>> list(d.items())
Georg Brandl116aa622007-08-15 14:28:22 +0000641 [('blue', [2, 4]), ('red', [1]), ('yellow', [1, 3])]
642
643Setting the :attr:`default_factory` to :class:`int` makes the
644:class:`defaultdict` useful for counting (like a bag or multiset in other
Christian Heimesfe337bf2008-03-23 21:54:12 +0000645languages):
Georg Brandl116aa622007-08-15 14:28:22 +0000646
647 >>> s = 'mississippi'
648 >>> d = defaultdict(int)
649 >>> for k in s:
650 ... d[k] += 1
651 ...
Ezio Melottic53a8942009-09-12 01:52:05 +0000652 >>> list(d.items())
Georg Brandl116aa622007-08-15 14:28:22 +0000653 [('i', 4), ('p', 2), ('s', 4), ('m', 1)]
654
655When a letter is first encountered, it is missing from the mapping, so the
656:attr:`default_factory` function calls :func:`int` to supply a default count of
657zero. The increment operation then builds up the count for each letter.
658
659The function :func:`int` which always returns zero is just a special case of
660constant functions. A faster and more flexible way to create constant functions
661is to use a lambda function which can supply any constant value (not just
Christian Heimesfe337bf2008-03-23 21:54:12 +0000662zero):
Georg Brandl116aa622007-08-15 14:28:22 +0000663
664 >>> def constant_factory(value):
665 ... return lambda: value
666 >>> d = defaultdict(constant_factory('<missing>'))
667 >>> d.update(name='John', action='ran')
668 >>> '%(name)s %(action)s to %(object)s' % d
669 'John ran to <missing>'
670
671Setting the :attr:`default_factory` to :class:`set` makes the
Christian Heimesfe337bf2008-03-23 21:54:12 +0000672:class:`defaultdict` useful for building a dictionary of sets:
Georg Brandl116aa622007-08-15 14:28:22 +0000673
674 >>> s = [('red', 1), ('blue', 2), ('red', 3), ('blue', 4), ('red', 1), ('blue', 4)]
675 >>> d = defaultdict(set)
676 >>> for k, v in s:
677 ... d[k].add(v)
678 ...
Ezio Melottic53a8942009-09-12 01:52:05 +0000679 >>> list(d.items())
Raymond Hettingerba7b5602011-03-22 22:57:49 -0700680 [('blue', {2, 4}), ('red', {1, 3})]
Georg Brandl116aa622007-08-15 14:28:22 +0000681
682
Guido van Rossum3d392eb2007-11-16 00:35:22 +0000683:func:`namedtuple` Factory Function for Tuples with Named Fields
Christian Heimes790c8232008-01-07 21:14:23 +0000684----------------------------------------------------------------
Georg Brandl116aa622007-08-15 14:28:22 +0000685
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000686Named tuples assign meaning to each position in a tuple and allow for more readable,
687self-documenting code. They can be used wherever regular tuples are used, and
688they add the ability to access fields by name instead of position index.
Georg Brandl116aa622007-08-15 14:28:22 +0000689
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000690.. function:: namedtuple(typename, field_names, verbose=False, rename=False)
Georg Brandl116aa622007-08-15 14:28:22 +0000691
692 Returns a new tuple subclass named *typename*. The new subclass is used to
Christian Heimesc3f30c42008-02-22 16:37:40 +0000693 create tuple-like objects that have fields accessible by attribute lookup as
Georg Brandl116aa622007-08-15 14:28:22 +0000694 well as being indexable and iterable. Instances of the subclass also have a
Benjamin Peterson4469d0c2008-11-30 22:46:23 +0000695 helpful docstring (with typename and field_names) and a helpful :meth:`__repr__`
Georg Brandl116aa622007-08-15 14:28:22 +0000696 method which lists the tuple contents in a ``name=value`` format.
697
Benjamin Peterson4469d0c2008-11-30 22:46:23 +0000698 The *field_names* are a single string with each fieldname separated by whitespace
699 and/or commas, for example ``'x y'`` or ``'x, y'``. Alternatively, *field_names*
Christian Heimes25bb7832008-01-11 16:17:00 +0000700 can be a sequence of strings such as ``['x', 'y']``.
Georg Brandl9afde1c2007-11-01 20:32:30 +0000701
702 Any valid Python identifier may be used for a fieldname except for names
Christian Heimes0449f632007-12-15 01:27:15 +0000703 starting with an underscore. Valid identifiers consist of letters, digits,
704 and underscores but do not start with a digit or underscore and cannot be
Georg Brandlf6945182008-02-01 11:56:49 +0000705 a :mod:`keyword` such as *class*, *for*, *return*, *global*, *pass*,
Georg Brandl9afde1c2007-11-01 20:32:30 +0000706 or *raise*.
Georg Brandl116aa622007-08-15 14:28:22 +0000707
Benjamin Petersona86f2c02009-02-10 02:41:10 +0000708 If *rename* is true, invalid fieldnames are automatically replaced
709 with positional names. For example, ``['abc', 'def', 'ghi', 'abc']`` is
Raymond Hettinger85737b82009-04-02 22:37:59 +0000710 converted to ``['abc', '_1', 'ghi', '_3']``, eliminating the keyword
Benjamin Petersona86f2c02009-02-10 02:41:10 +0000711 ``def`` and the duplicate fieldname ``abc``.
712
Raymond Hettinger2ebea412011-03-23 12:52:23 -0700713 If *verbose* is true, the class definition is printed after it is
714 built. This option is outdated; instead, it is simpler to print the
715 :attr:`_source` attribute.
Georg Brandl116aa622007-08-15 14:28:22 +0000716
Georg Brandl9afde1c2007-11-01 20:32:30 +0000717 Named tuple instances do not have per-instance dictionaries, so they are
Thomas Wouters8ce81f72007-09-20 18:22:40 +0000718 lightweight and require no more memory than regular tuples.
Georg Brandl116aa622007-08-15 14:28:22 +0000719
Raymond Hettingerb62ad242009-03-02 22:16:43 +0000720 .. versionchanged:: 3.1
Georg Brandl67b21b72010-08-17 15:07:14 +0000721 Added support for *rename*.
Benjamin Petersona86f2c02009-02-10 02:41:10 +0000722
Christian Heimesfe337bf2008-03-23 21:54:12 +0000723
724.. doctest::
725 :options: +NORMALIZE_WHITESPACE
Georg Brandl116aa622007-08-15 14:28:22 +0000726
Raymond Hettinger0ef956f2010-11-21 23:23:29 +0000727 >>> # Basic example
Raymond Hettinger15aded82011-03-15 17:25:51 -0700728 >>> Point = namedtuple('Point', ['x', 'y'])
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000729 >>> p = Point(11, y=22) # instantiate with positional or keyword arguments
Christian Heimes99170a52007-12-19 02:07:34 +0000730 >>> p[0] + p[1] # indexable like the plain tuple (11, 22)
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000731 33
732 >>> x, y = p # unpack like a regular tuple
733 >>> x, y
734 (11, 22)
Christian Heimesc3f30c42008-02-22 16:37:40 +0000735 >>> p.x + p.y # fields also accessible by name
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000736 33
737 >>> p # readable __repr__ with a name=value style
738 Point(x=11, y=22)
Georg Brandl116aa622007-08-15 14:28:22 +0000739
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000740Named tuples are especially useful for assigning field names to result tuples returned
741by the :mod:`csv` or :mod:`sqlite3` modules::
742
Guido van Rossum3d392eb2007-11-16 00:35:22 +0000743 EmployeeRecord = namedtuple('EmployeeRecord', 'name, age, title, department, paygrade')
Georg Brandl9afde1c2007-11-01 20:32:30 +0000744
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000745 import csv
Christian Heimesfaf2f632008-01-06 16:59:19 +0000746 for emp in map(EmployeeRecord._make, csv.reader(open("employees.csv", "rb"))):
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000747 print(emp.name, emp.title)
748
Georg Brandl9afde1c2007-11-01 20:32:30 +0000749 import sqlite3
750 conn = sqlite3.connect('/companydata')
751 cursor = conn.cursor()
752 cursor.execute('SELECT name, age, title, department, paygrade FROM employees')
Christian Heimesfaf2f632008-01-06 16:59:19 +0000753 for emp in map(EmployeeRecord._make, cursor.fetchall()):
Christian Heimes00412232008-01-10 16:02:19 +0000754 print(emp.name, emp.title)
Georg Brandl9afde1c2007-11-01 20:32:30 +0000755
Christian Heimes99170a52007-12-19 02:07:34 +0000756In addition to the methods inherited from tuples, named tuples support
Raymond Hettinger2ebea412011-03-23 12:52:23 -0700757three additional methods and two attributes. To prevent conflicts with
Christian Heimes2380ac72008-01-09 00:17:24 +0000758field names, the method and attribute names start with an underscore.
Christian Heimes99170a52007-12-19 02:07:34 +0000759
Benjamin Peterson0b9fb802010-07-18 14:23:36 +0000760.. classmethod:: somenamedtuple._make(iterable)
Christian Heimes99170a52007-12-19 02:07:34 +0000761
Christian Heimesfaf2f632008-01-06 16:59:19 +0000762 Class method that makes a new instance from an existing sequence or iterable.
Christian Heimes99170a52007-12-19 02:07:34 +0000763
Christian Heimesfe337bf2008-03-23 21:54:12 +0000764.. doctest::
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000765
Christian Heimesfaf2f632008-01-06 16:59:19 +0000766 >>> t = [11, 22]
767 >>> Point._make(t)
768 Point(x=11, y=22)
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000769
Christian Heimes790c8232008-01-07 21:14:23 +0000770.. method:: somenamedtuple._asdict()
Georg Brandl9afde1c2007-11-01 20:32:30 +0000771
Raymond Hettingera4f52b12009-03-02 22:28:31 +0000772 Return a new :class:`OrderedDict` which maps field names to their corresponding
773 values::
Georg Brandl9afde1c2007-11-01 20:32:30 +0000774
Christian Heimes0449f632007-12-15 01:27:15 +0000775 >>> p._asdict()
Raymond Hettingera4f52b12009-03-02 22:28:31 +0000776 OrderedDict([('x', 11), ('y', 22)])
777
Raymond Hettingera88e4da2009-03-03 05:12:27 +0000778 .. versionchanged:: 3.1
Raymond Hettingera4f52b12009-03-02 22:28:31 +0000779 Returns an :class:`OrderedDict` instead of a regular :class:`dict`.
Christian Heimesfe337bf2008-03-23 21:54:12 +0000780
Christian Heimes790c8232008-01-07 21:14:23 +0000781.. method:: somenamedtuple._replace(kwargs)
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000782
Christian Heimesfe337bf2008-03-23 21:54:12 +0000783 Return a new instance of the named tuple replacing specified fields with new
784 values:
Thomas Wouters8ce81f72007-09-20 18:22:40 +0000785
786::
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000787
788 >>> p = Point(x=11, y=22)
Christian Heimes0449f632007-12-15 01:27:15 +0000789 >>> p._replace(x=33)
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000790 Point(x=33, y=22)
791
Guido van Rossum3d392eb2007-11-16 00:35:22 +0000792 >>> for partnum, record in inventory.items():
Christian Heimes454f37b2008-01-10 00:10:02 +0000793 ... inventory[partnum] = record._replace(price=newprices[partnum], timestamp=time.now())
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000794
Raymond Hettinger2ebea412011-03-23 12:52:23 -0700795.. attribute:: somenamedtuple._source
796
797 A string with the pure Python source code used to create the named
798 tuple class. The source makes the named tuple self-documenting.
799 It can be printed, executed using :func:`exec`, or saved to a file
800 and imported.
801
802 .. versionadded:: 3.3
803
Christian Heimes790c8232008-01-07 21:14:23 +0000804.. attribute:: somenamedtuple._fields
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000805
Christian Heimes2380ac72008-01-09 00:17:24 +0000806 Tuple of strings listing the field names. Useful for introspection
Georg Brandl9afde1c2007-11-01 20:32:30 +0000807 and for creating new named tuple types from existing named tuples.
Thomas Wouters8ce81f72007-09-20 18:22:40 +0000808
Christian Heimesfe337bf2008-03-23 21:54:12 +0000809.. doctest::
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000810
Christian Heimes0449f632007-12-15 01:27:15 +0000811 >>> p._fields # view the field names
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000812 ('x', 'y')
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000813
Guido van Rossum3d392eb2007-11-16 00:35:22 +0000814 >>> Color = namedtuple('Color', 'red green blue')
Christian Heimes0449f632007-12-15 01:27:15 +0000815 >>> Pixel = namedtuple('Pixel', Point._fields + Color._fields)
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000816 >>> Pixel(11, 22, 128, 255, 0)
Christian Heimes454f37b2008-01-10 00:10:02 +0000817 Pixel(x=11, y=22, red=128, green=255, blue=0)
Georg Brandl116aa622007-08-15 14:28:22 +0000818
Christian Heimes0449f632007-12-15 01:27:15 +0000819To retrieve a field whose name is stored in a string, use the :func:`getattr`
Christian Heimesfe337bf2008-03-23 21:54:12 +0000820function:
Christian Heimes0449f632007-12-15 01:27:15 +0000821
822 >>> getattr(p, 'x')
823 11
824
Raymond Hettinger651453a2009-02-11 00:20:02 +0000825To convert a dictionary to a named tuple, use the double-star-operator
826(as described in :ref:`tut-unpacking-arguments`):
Christian Heimes99170a52007-12-19 02:07:34 +0000827
828 >>> d = {'x': 11, 'y': 22}
829 >>> Point(**d)
830 Point(x=11, y=22)
831
Guido van Rossum3d392eb2007-11-16 00:35:22 +0000832Since a named tuple is a regular Python class, it is easy to add or change
Christian Heimes043d6f62008-01-07 17:19:16 +0000833functionality with a subclass. Here is how to add a calculated field and
Christian Heimesfe337bf2008-03-23 21:54:12 +0000834a fixed-width print format:
Guido van Rossum3d392eb2007-11-16 00:35:22 +0000835
Christian Heimes043d6f62008-01-07 17:19:16 +0000836 >>> class Point(namedtuple('Point', 'x y')):
Raymond Hettinger15aded82011-03-15 17:25:51 -0700837 __slots__ = ()
838 @property
839 def hypot(self):
840 return (self.x ** 2 + self.y ** 2) ** 0.5
841 def __str__(self):
842 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 +0000843
Georg Brandl0df79792008-10-04 18:33:26 +0000844 >>> for p in Point(3, 4), Point(14, 5/7):
Raymond Hettinger15aded82011-03-15 17:25:51 -0700845 print(p)
Christian Heimes25bb7832008-01-11 16:17:00 +0000846 Point: x= 3.000 y= 4.000 hypot= 5.000
847 Point: x=14.000 y= 0.714 hypot=14.018
Christian Heimes043d6f62008-01-07 17:19:16 +0000848
Georg Brandlaf5c2382009-12-28 08:02:38 +0000849The subclass shown above sets ``__slots__`` to an empty tuple. This helps
Christian Heimes679db4a2008-01-18 09:56:22 +0000850keep memory requirements low by preventing the creation of instance dictionaries.
851
Christian Heimes2380ac72008-01-09 00:17:24 +0000852Subclassing is not useful for adding new, stored fields. Instead, simply
Christian Heimesfe337bf2008-03-23 21:54:12 +0000853create a new named tuple type from the :attr:`_fields` attribute:
Christian Heimes2380ac72008-01-09 00:17:24 +0000854
Christian Heimes25bb7832008-01-11 16:17:00 +0000855 >>> Point3D = namedtuple('Point3D', Point._fields + ('z',))
Christian Heimes2380ac72008-01-09 00:17:24 +0000856
857Default values can be implemented by using :meth:`_replace` to
Christian Heimesfe337bf2008-03-23 21:54:12 +0000858customize a prototype instance:
Guido van Rossum3d392eb2007-11-16 00:35:22 +0000859
860 >>> Account = namedtuple('Account', 'owner balance transaction_count')
Christian Heimes587c2bf2008-01-19 16:21:02 +0000861 >>> default_account = Account('<owner name>', 0.0, 0)
862 >>> johns_account = default_account._replace(owner='John')
Raymond Hettingerb2d09452011-03-22 22:36:21 -0700863 >>> janes_account = default_account._replace(owner='Jane')
Guido van Rossum3d392eb2007-11-16 00:35:22 +0000864
Christian Heimese4ca8152008-05-08 17:18:53 +0000865Enumerated constants can be implemented with named tuples, but it is simpler
866and more efficient to use a simple class declaration:
867
868 >>> Status = namedtuple('Status', 'open pending closed')._make(range(3))
869 >>> Status.open, Status.pending, Status.closed
870 (0, 1, 2)
871 >>> class Status:
Raymond Hettinger15aded82011-03-15 17:25:51 -0700872 open, pending, closed = range(3)
Christian Heimese4ca8152008-05-08 17:18:53 +0000873
Raymond Hettinger651453a2009-02-11 00:20:02 +0000874.. seealso::
Thomas Wouters47b49bf2007-08-30 22:15:33 +0000875
Raymond Hettinger6c94e6f2011-03-31 15:46:06 -0700876 * `Named tuple recipe <http://code.activestate.com/recipes/500261/>`_
877 adapted for Python 2.4.
878
879 * `Recipe for named tuple abstract base class with a metaclass mix-in
880 <http://code.activestate.com/recipes/577629-namedtupleabc-abstract-base-class-mix-in-for-named/>`_
881 by Jan Kaliszewski. Besides providing an :term:`abstract base class` for
882 named tuples, it also supports an alternate :term:`metaclass`-based
883 constructor that is convenient for use cases where named tuples are being
884 subclassed.
Raymond Hettingere4c96ad2008-02-06 01:23:58 +0000885
886
Raymond Hettinger2d32f632009-03-02 21:24:57 +0000887:class:`OrderedDict` objects
888----------------------------
889
890Ordered dictionaries are just like regular dictionaries but they remember the
891order that items were inserted. When iterating over an ordered dictionary,
892the items are returned in the order their keys were first added.
893
894.. class:: OrderedDict([items])
895
896 Return an instance of a dict subclass, supporting the usual :class:`dict`
897 methods. An *OrderedDict* is a dict that remembers the order that keys
898 were first inserted. If a new entry overwrites an existing entry, the
899 original insertion position is left unchanged. Deleting an entry and
900 reinserting it will move it to the end.
901
Benjamin Petersond45bf582009-03-02 21:44:54 +0000902 .. versionadded:: 3.1
Raymond Hettinger2d32f632009-03-02 21:24:57 +0000903
Benjamin Petersond319ad52010-07-18 14:27:02 +0000904 .. method:: popitem(last=True)
Raymond Hettingerdc879f02009-03-19 20:30:56 +0000905
Benjamin Petersond319ad52010-07-18 14:27:02 +0000906 The :meth:`popitem` method for ordered dictionaries returns and removes a
907 (key, value) pair. The pairs are returned in LIFO order if *last* is true
908 or FIFO order if false.
Raymond Hettinger2d32f632009-03-02 21:24:57 +0000909
Raymond Hettingerf45abc92010-09-06 21:26:09 +0000910 .. method:: move_to_end(key, last=True)
911
912 Move an existing *key* to either end of an ordered dictionary. The item
913 is moved to the right end if *last* is true (the default) or to the
914 beginning if *last* is false. Raises :exc:`KeyError` if the *key* does
915 not exist::
916
917 >>> d = OrderedDict.fromkeys('abcde')
918 >>> d.move_to_end('b')
Raymond Hettinger4d5208d2011-06-25 11:39:00 +0200919 >>> ''.join(d.keys())
Raymond Hettingerf45abc92010-09-06 21:26:09 +0000920 'acdeb'
Éric Araujo1cb25aa2010-11-06 07:03:07 +0000921 >>> d.move_to_end('b', last=False)
Raymond Hettinger4d5208d2011-06-25 11:39:00 +0200922 >>> ''.join(d.keys())
Raymond Hettingerf45abc92010-09-06 21:26:09 +0000923 'bacde'
924
925 .. versionadded:: 3.2
926
Raymond Hettingere9091502009-05-19 17:40:07 +0000927In addition to the usual mapping methods, ordered dictionaries also support
928reverse iteration using :func:`reversed`.
929
Raymond Hettinger2d32f632009-03-02 21:24:57 +0000930Equality tests between :class:`OrderedDict` objects are order-sensitive
931and are implemented as ``list(od1.items())==list(od2.items())``.
932Equality tests between :class:`OrderedDict` objects and other
933:class:`Mapping` objects are order-insensitive like regular dictionaries.
934This allows :class:`OrderedDict` objects to be substituted anywhere a
935regular dictionary is used.
936
Raymond Hettinger36180782009-04-09 22:34:23 +0000937The :class:`OrderedDict` constructor and :meth:`update` method both accept
938keyword arguments, but their order is lost because Python's function call
939semantics pass-in keyword arguments using a regular unordered dictionary.
940
Raymond Hettingerdc879f02009-03-19 20:30:56 +0000941.. seealso::
942
943 `Equivalent OrderedDict recipe <http://code.activestate.com/recipes/576693/>`_
944 that runs on Python 2.4 or later.
945
Raymond Hettinger7bba6832011-04-15 17:43:19 -0700946:class:`OrderedDict` Examples and Recipes
947^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
948
Raymond Hettinger0e312012009-11-10 18:35:46 +0000949Since an ordered dictionary remembers its insertion order, it can be used
950in conjuction with sorting to make a sorted dictionary::
951
952 >>> # regular unsorted dictionary
953 >>> d = {'banana': 3, 'apple':4, 'pear': 1, 'orange': 2}
954
955 >>> # dictionary sorted by key
956 >>> OrderedDict(sorted(d.items(), key=lambda t: t[0]))
957 OrderedDict([('apple', 4), ('banana', 3), ('orange', 2), ('pear', 1)])
958
959 >>> # dictionary sorted by value
960 >>> OrderedDict(sorted(d.items(), key=lambda t: t[1]))
961 OrderedDict([('pear', 1), ('orange', 2), ('banana', 3), ('apple', 4)])
962
963 >>> # dictionary sorted by length of the key string
964 >>> OrderedDict(sorted(d.items(), key=lambda t: len(t[0])))
965 OrderedDict([('pear', 1), ('apple', 4), ('orange', 2), ('banana', 3)])
966
967The new sorted dictionaries maintain their sort order when entries
968are deleted. But when new keys are added, the keys are appended
969to the end and the sort is not maintained.
970
Raymond Hettinger4821ef82010-07-31 10:14:41 +0000971It is also straight-forward to create an ordered dictionary variant
972that the remembers the order the keys were *last* inserted.
973If a new entry overwrites an existing entry, the
974original insertion position is changed and moved to the end::
975
976 class LastUpdatedOrderedDict(OrderedDict):
Georg Brandl77570e22010-12-18 16:21:58 +0000977 'Store items in the order the keys were last added'
Raymond Hettinger7bba6832011-04-15 17:43:19 -0700978
Raymond Hettinger4821ef82010-07-31 10:14:41 +0000979 def __setitem__(self, key, value):
980 if key in self:
981 del self[key]
982 OrderedDict.__setitem__(self, key, value)
983
Éric Araujo889a7dc2011-08-19 00:40:46 +0200984An ordered dictionary can be combined with the :class:`Counter` class
Raymond Hettinger7bba6832011-04-15 17:43:19 -0700985so that the counter remembers the order elements are first encountered::
986
987 class OrderedCounter(Counter, OrderedDict):
988 'Counter that remembers the order elements are first encountered'
989
Raymond Hettinger7bba6832011-04-15 17:43:19 -0700990 def __repr__(self):
991 return '%s(%r)' % (self.__class__.__name__, OrderedDict(self))
992
993 def __reduce__(self):
994 return self.__class__, (OrderedDict(self),)
995
Raymond Hettingere4c96ad2008-02-06 01:23:58 +0000996
997:class:`UserDict` objects
Mark Summerfield8f2d0062008-02-06 13:30:44 +0000998-------------------------
Raymond Hettingere4c96ad2008-02-06 01:23:58 +0000999
Georg Brandl48310cd2009-01-03 21:18:54 +00001000The class, :class:`UserDict` acts as a wrapper around dictionary objects.
1001The need for this class has been partially supplanted by the ability to
Raymond Hettingere4c96ad2008-02-06 01:23:58 +00001002subclass directly from :class:`dict`; however, this class can be easier
1003to work with because the underlying dictionary is accessible as an
1004attribute.
1005
1006.. class:: UserDict([initialdata])
1007
1008 Class that simulates a dictionary. The instance's contents are kept in a
1009 regular dictionary, which is accessible via the :attr:`data` attribute of
1010 :class:`UserDict` instances. If *initialdata* is provided, :attr:`data` is
1011 initialized with its contents; note that a reference to *initialdata* will not
1012 be kept, allowing it be used for other purposes.
1013
Benjamin Petersond319ad52010-07-18 14:27:02 +00001014 In addition to supporting the methods and operations of mappings,
1015 :class:`UserDict` instances provide the following attribute:
Raymond Hettingere4c96ad2008-02-06 01:23:58 +00001016
Benjamin Petersond319ad52010-07-18 14:27:02 +00001017 .. attribute:: data
Raymond Hettingere4c96ad2008-02-06 01:23:58 +00001018
Benjamin Petersond319ad52010-07-18 14:27:02 +00001019 A real dictionary used to store the contents of the :class:`UserDict`
1020 class.
Raymond Hettinger53dbe392008-02-12 20:03:09 +00001021
1022
1023
1024:class:`UserList` objects
1025-------------------------
1026
1027This class acts as a wrapper around list objects. It is a useful base class
Georg Brandl48310cd2009-01-03 21:18:54 +00001028for your own list-like classes which can inherit from them and override
Raymond Hettinger53dbe392008-02-12 20:03:09 +00001029existing methods or add new ones. In this way, one can add new behaviors to
1030lists.
1031
Georg Brandl48310cd2009-01-03 21:18:54 +00001032The need for this class has been partially supplanted by the ability to
Raymond Hettinger53dbe392008-02-12 20:03:09 +00001033subclass directly from :class:`list`; however, this class can be easier
1034to work with because the underlying list is accessible as an attribute.
1035
1036.. class:: UserList([list])
1037
1038 Class that simulates a list. The instance's contents are kept in a regular
1039 list, which is accessible via the :attr:`data` attribute of :class:`UserList`
1040 instances. The instance's contents are initially set to a copy of *list*,
1041 defaulting to the empty list ``[]``. *list* can be any iterable, for
1042 example a real Python list or a :class:`UserList` object.
1043
Benjamin Petersond319ad52010-07-18 14:27:02 +00001044 In addition to supporting the methods and operations of mutable sequences,
1045 :class:`UserList` instances provide the following attribute:
Raymond Hettinger53dbe392008-02-12 20:03:09 +00001046
Benjamin Petersond319ad52010-07-18 14:27:02 +00001047 .. attribute:: data
Raymond Hettinger53dbe392008-02-12 20:03:09 +00001048
Benjamin Petersond319ad52010-07-18 14:27:02 +00001049 A real :class:`list` object used to store the contents of the
1050 :class:`UserList` class.
Raymond Hettinger53dbe392008-02-12 20:03:09 +00001051
1052**Subclassing requirements:** Subclasses of :class:`UserList` are expect to
1053offer a constructor which can be called with either no arguments or one
1054argument. List operations which return a new sequence attempt to create an
1055instance of the actual implementation class. To do so, it assumes that the
1056constructor can be called with a single parameter, which is a sequence object
1057used as a data source.
1058
1059If a derived class does not wish to comply with this requirement, all of the
1060special methods supported by this class will need to be overridden; please
1061consult the sources for information about the methods which need to be provided
1062in that case.
Raymond Hettingerb3a65f82008-02-21 22:11:37 +00001063
1064:class:`UserString` objects
Christian Heimesc3f30c42008-02-22 16:37:40 +00001065---------------------------
Raymond Hettingerb3a65f82008-02-21 22:11:37 +00001066
Georg Brandl48310cd2009-01-03 21:18:54 +00001067The class, :class:`UserString` acts as a wrapper around string objects.
1068The need for this class has been partially supplanted by the ability to
Raymond Hettingerb3a65f82008-02-21 22:11:37 +00001069subclass directly from :class:`str`; however, this class can be easier
1070to work with because the underlying string is accessible as an
1071attribute.
1072
1073.. class:: UserString([sequence])
1074
1075 Class that simulates a string or a Unicode string object. The instance's
Georg Brandl48310cd2009-01-03 21:18:54 +00001076 content is kept in a regular string object, which is accessible via the
1077 :attr:`data` attribute of :class:`UserString` instances. The instance's
Raymond Hettingerb3a65f82008-02-21 22:11:37 +00001078 contents are initially set to a copy of *sequence*. The *sequence* can
1079 be an instance of :class:`bytes`, :class:`str`, :class:`UserString` (or a
1080 subclass) or an arbitrary sequence which can be converted into a string using
1081 the built-in :func:`str` function.