blob: 42c867d8b0a4f3a3b6a57eb911b18dec59411ab3 [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
36 Moved :ref:`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
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
62 A class:`ChainMap` incorporates the underlying mappings by reference. So, if
63 one of the underlying mappings gets updated, those changes will be reflected
64 in class:`ChainMap`.
65
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
74 modified to change which mappings are searched. The list should
75 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
85 .. attribute:: parents()
86
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
136 <https://github.com/enthought/codetools>`_\ has options to support
137 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
152 <http://code.activestate.com/recipes/305268/>`_\.
153
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
266 c += Counter() # 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 Hettinger22f18852010-04-12 21:45:14 +0000286.. note::
287
288 Counters were primarily designed to work with positive integers to represent
289 running counts; however, care was taken to not unnecessarily preclude use
290 cases needing other types or negative values. To help with those use cases,
291 this section documents the minimum range and type restrictions.
292
293 * The :class:`Counter` class itself is a dictionary subclass with no
294 restrictions on its keys and values. The values are intended to be numbers
295 representing counts, but you *could* store anything in the value field.
296
297 * The :meth:`most_common` method requires only that the values be orderable.
298
299 * For in-place operations such as ``c[key] += 1``, the value type need only
300 support addition and subtraction. So fractions, floats, and decimals would
301 work and negative values are supported. The same is also true for
302 :meth:`update` and :meth:`subtract` which allow negative and zero values
303 for both inputs and outputs.
304
305 * The multiset methods are designed only for use cases with positive values.
306 The inputs may be negative or zero, but only outputs with positive values
307 are created. There are no type restrictions, but the value type needs to
308 support support addition, subtraction, and comparison.
309
310 * The :meth:`elements` method requires integer counts. It ignores zero and
311 negative counts.
312
Raymond Hettingerb14043c2009-01-20 23:44:31 +0000313.. seealso::
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000314
Raymond Hettinger94adc8e2009-01-22 05:27:37 +0000315 * `Counter class <http://code.activestate.com/recipes/576611/>`_
316 adapted for Python 2.5 and an early `Bag recipe
317 <http://code.activestate.com/recipes/259174/>`_ for Python 2.4.
318
Raymond Hettingerb14043c2009-01-20 23:44:31 +0000319 * `Bag class <http://www.gnu.org/software/smalltalk/manual-base/html_node/Bag.html>`_
320 in Smalltalk.
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000321
Raymond Hettingerb14043c2009-01-20 23:44:31 +0000322 * Wikipedia entry for `Multisets <http://en.wikipedia.org/wiki/Multiset>`_\.
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000323
Raymond Hettingerb14043c2009-01-20 23:44:31 +0000324 * `C++ multisets <http://www.demo2s.com/Tutorial/Cpp/0380__set-multiset/Catalog0380__set-multiset.htm>`_
Raymond Hettinger94adc8e2009-01-22 05:27:37 +0000325 tutorial with examples.
Raymond Hettingerb14043c2009-01-20 23:44:31 +0000326
Raymond Hettinger94adc8e2009-01-22 05:27:37 +0000327 * For mathematical operations on multisets and their use cases, see
Raymond Hettingerb14043c2009-01-20 23:44:31 +0000328 *Knuth, Donald. The Art of Computer Programming Volume II,
329 Section 4.6.3, Exercise 19*\.
330
Raymond Hettinger670eaec2009-01-21 23:14:07 +0000331 * To enumerate all distinct multisets of a given size over a given set of
Raymond Hettingerd07d9392009-01-27 04:20:44 +0000332 elements, see :func:`itertools.combinations_with_replacement`.
Raymond Hettingerb14043c2009-01-20 23:44:31 +0000333
Raymond Hettinger94adc8e2009-01-22 05:27:37 +0000334 map(Counter, combinations_with_replacement('ABC', 2)) --> AA AB AC BB BC CC
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000335
336
Georg Brandl116aa622007-08-15 14:28:22 +0000337:class:`deque` objects
338----------------------
339
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000340.. class:: deque([iterable, [maxlen]])
Georg Brandl116aa622007-08-15 14:28:22 +0000341
342 Returns a new deque object initialized left-to-right (using :meth:`append`) with
343 data from *iterable*. If *iterable* is not specified, the new deque is empty.
344
345 Deques are a generalization of stacks and queues (the name is pronounced "deck"
346 and is short for "double-ended queue"). Deques support thread-safe, memory
347 efficient appends and pops from either side of the deque with approximately the
348 same O(1) performance in either direction.
349
350 Though :class:`list` objects support similar operations, they are optimized for
351 fast fixed-length operations and incur O(n) memory movement costs for
352 ``pop(0)`` and ``insert(0, v)`` operations which change both the size and
353 position of the underlying data representation.
354
Georg Brandl116aa622007-08-15 14:28:22 +0000355
Georg Brandl9afde1c2007-11-01 20:32:30 +0000356 If *maxlen* is not specified or is *None*, deques may grow to an
357 arbitrary length. Otherwise, the deque is bounded to the specified maximum
358 length. Once a bounded length deque is full, when new items are added, a
359 corresponding number of items are discarded from the opposite end. Bounded
360 length deques provide functionality similar to the ``tail`` filter in
361 Unix. They are also useful for tracking transactions and other pools of data
362 where only the most recent activity is of interest.
363
Georg Brandl9afde1c2007-11-01 20:32:30 +0000364
Benjamin Petersone41251e2008-04-25 01:59:09 +0000365 Deque objects support the following methods:
Georg Brandl116aa622007-08-15 14:28:22 +0000366
Benjamin Petersone41251e2008-04-25 01:59:09 +0000367 .. method:: append(x)
Georg Brandl116aa622007-08-15 14:28:22 +0000368
Benjamin Petersone41251e2008-04-25 01:59:09 +0000369 Add *x* to the right side of the deque.
Georg Brandl116aa622007-08-15 14:28:22 +0000370
371
Benjamin Petersone41251e2008-04-25 01:59:09 +0000372 .. method:: appendleft(x)
Georg Brandl116aa622007-08-15 14:28:22 +0000373
Benjamin Petersone41251e2008-04-25 01:59:09 +0000374 Add *x* to the left side of the deque.
Georg Brandl116aa622007-08-15 14:28:22 +0000375
376
Benjamin Petersone41251e2008-04-25 01:59:09 +0000377 .. method:: clear()
Georg Brandl116aa622007-08-15 14:28:22 +0000378
Benjamin Petersone41251e2008-04-25 01:59:09 +0000379 Remove all elements from the deque leaving it with length 0.
Georg Brandl116aa622007-08-15 14:28:22 +0000380
381
Raymond Hettinger44459de2010-04-03 23:20:46 +0000382 .. method:: count(x)
383
384 Count the number of deque elements equal to *x*.
385
386 .. versionadded:: 3.2
387
Georg Brandl67b21b72010-08-17 15:07:14 +0000388
Benjamin Petersone41251e2008-04-25 01:59:09 +0000389 .. method:: extend(iterable)
Georg Brandl116aa622007-08-15 14:28:22 +0000390
Benjamin Petersone41251e2008-04-25 01:59:09 +0000391 Extend the right side of the deque by appending elements from the iterable
392 argument.
Georg Brandl116aa622007-08-15 14:28:22 +0000393
394
Benjamin Petersone41251e2008-04-25 01:59:09 +0000395 .. method:: extendleft(iterable)
Georg Brandl116aa622007-08-15 14:28:22 +0000396
Benjamin Petersone41251e2008-04-25 01:59:09 +0000397 Extend the left side of the deque by appending elements from *iterable*.
398 Note, the series of left appends results in reversing the order of
399 elements in the iterable argument.
Georg Brandl116aa622007-08-15 14:28:22 +0000400
401
Benjamin Petersone41251e2008-04-25 01:59:09 +0000402 .. method:: pop()
Georg Brandl116aa622007-08-15 14:28:22 +0000403
Benjamin Petersone41251e2008-04-25 01:59:09 +0000404 Remove and return an element from the right side of the deque. If no
405 elements are present, raises an :exc:`IndexError`.
Georg Brandl116aa622007-08-15 14:28:22 +0000406
407
Benjamin Petersone41251e2008-04-25 01:59:09 +0000408 .. method:: popleft()
Georg Brandl116aa622007-08-15 14:28:22 +0000409
Benjamin Petersone41251e2008-04-25 01:59:09 +0000410 Remove and return an element from the left side of the deque. If no
411 elements are present, raises an :exc:`IndexError`.
Georg Brandl116aa622007-08-15 14:28:22 +0000412
413
Benjamin Petersone41251e2008-04-25 01:59:09 +0000414 .. method:: remove(value)
Georg Brandl116aa622007-08-15 14:28:22 +0000415
Benjamin Petersone41251e2008-04-25 01:59:09 +0000416 Removed the first occurrence of *value*. If not found, raises a
417 :exc:`ValueError`.
Georg Brandl116aa622007-08-15 14:28:22 +0000418
Georg Brandl67b21b72010-08-17 15:07:14 +0000419
Raymond Hettingere5fdedb2009-12-10 00:47:21 +0000420 .. method:: reverse()
421
422 Reverse the elements of the deque in-place and then return ``None``.
423
424 .. versionadded:: 3.2
Georg Brandl116aa622007-08-15 14:28:22 +0000425
Georg Brandl67b21b72010-08-17 15:07:14 +0000426
Benjamin Petersone41251e2008-04-25 01:59:09 +0000427 .. method:: rotate(n)
Georg Brandl116aa622007-08-15 14:28:22 +0000428
Benjamin Petersone41251e2008-04-25 01:59:09 +0000429 Rotate the deque *n* steps to the right. If *n* is negative, rotate to
430 the left. Rotating one step to the right is equivalent to:
431 ``d.appendleft(d.pop())``.
432
Georg Brandl116aa622007-08-15 14:28:22 +0000433
Raymond Hettinger5bb0f0e2009-03-10 12:56:32 +0000434 Deque objects also provide one read-only attribute:
435
436 .. attribute:: maxlen
437
438 Maximum size of a deque or *None* if unbounded.
439
Raymond Hettinger150fb9c2009-03-10 22:48:06 +0000440 .. versionadded:: 3.1
Raymond Hettinger5bb0f0e2009-03-10 12:56:32 +0000441
442
Georg Brandl116aa622007-08-15 14:28:22 +0000443In addition to the above, deques support iteration, pickling, ``len(d)``,
444``reversed(d)``, ``copy.copy(d)``, ``copy.deepcopy(d)``, membership testing with
Benjamin Peterson206e3072008-10-19 14:07:49 +0000445the :keyword:`in` operator, and subscript references such as ``d[-1]``. Indexed
446access is O(1) at both ends but slows to O(n) in the middle. For fast random
447access, use lists instead.
Georg Brandl116aa622007-08-15 14:28:22 +0000448
Christian Heimesfe337bf2008-03-23 21:54:12 +0000449Example:
450
451.. doctest::
Georg Brandl116aa622007-08-15 14:28:22 +0000452
453 >>> from collections import deque
454 >>> d = deque('ghi') # make a new deque with three items
455 >>> for elem in d: # iterate over the deque's elements
Neal Norwitz752abd02008-05-13 04:55:24 +0000456 ... print(elem.upper())
Georg Brandl116aa622007-08-15 14:28:22 +0000457 G
458 H
459 I
460
461 >>> d.append('j') # add a new entry to the right side
462 >>> d.appendleft('f') # add a new entry to the left side
463 >>> d # show the representation of the deque
464 deque(['f', 'g', 'h', 'i', 'j'])
465
466 >>> d.pop() # return and remove the rightmost item
467 'j'
468 >>> d.popleft() # return and remove the leftmost item
469 'f'
470 >>> list(d) # list the contents of the deque
471 ['g', 'h', 'i']
472 >>> d[0] # peek at leftmost item
473 'g'
474 >>> d[-1] # peek at rightmost item
475 'i'
476
477 >>> list(reversed(d)) # list the contents of a deque in reverse
478 ['i', 'h', 'g']
479 >>> 'h' in d # search the deque
480 True
481 >>> d.extend('jkl') # add multiple elements at once
482 >>> d
483 deque(['g', 'h', 'i', 'j', 'k', 'l'])
484 >>> d.rotate(1) # right rotation
485 >>> d
486 deque(['l', 'g', 'h', 'i', 'j', 'k'])
487 >>> d.rotate(-1) # left rotation
488 >>> d
489 deque(['g', 'h', 'i', 'j', 'k', 'l'])
490
491 >>> deque(reversed(d)) # make a new deque in reverse order
492 deque(['l', 'k', 'j', 'i', 'h', 'g'])
493 >>> d.clear() # empty the deque
494 >>> d.pop() # cannot pop from an empty deque
495 Traceback (most recent call last):
496 File "<pyshell#6>", line 1, in -toplevel-
497 d.pop()
498 IndexError: pop from an empty deque
499
500 >>> d.extendleft('abc') # extendleft() reverses the input order
501 >>> d
502 deque(['c', 'b', 'a'])
503
504
Georg Brandl9afde1c2007-11-01 20:32:30 +0000505:class:`deque` Recipes
506^^^^^^^^^^^^^^^^^^^^^^
Georg Brandl116aa622007-08-15 14:28:22 +0000507
508This section shows various approaches to working with deques.
509
Raymond Hettingerd2ee64d2009-03-31 22:52:48 +0000510Bounded length deques provide functionality similar to the ``tail`` filter
511in Unix::
512
513 def tail(filename, n=10):
514 'Return the last n lines of a file'
Éric Araujoa3dd56b2011-03-11 17:42:48 +0100515 with open(filename) as f:
516 return deque(f, n)
Raymond Hettingerd2ee64d2009-03-31 22:52:48 +0000517
518Another approach to using deques is to maintain a sequence of recently
519added elements by appending to the right and popping to the left::
520
521 def moving_average(iterable, n=3):
522 # moving_average([40, 30, 50, 46, 39, 44]) --> 40.0 42.0 45.0 43.0
523 # http://en.wikipedia.org/wiki/Moving_average
524 it = iter(iterable)
Raymond Hettingerd40285a2009-05-22 01:11:26 +0000525 d = deque(itertools.islice(it, n-1))
526 d.appendleft(0)
Raymond Hettingerd2ee64d2009-03-31 22:52:48 +0000527 s = sum(d)
Raymond Hettingerd2ee64d2009-03-31 22:52:48 +0000528 for elem in it:
529 s += elem - d.popleft()
530 d.append(elem)
531 yield s / n
532
Georg Brandl116aa622007-08-15 14:28:22 +0000533The :meth:`rotate` method provides a way to implement :class:`deque` slicing and
Ezio Melotti0639d5a2009-12-19 23:26:38 +0000534deletion. For example, a pure Python implementation of ``del d[n]`` relies on
Georg Brandl116aa622007-08-15 14:28:22 +0000535the :meth:`rotate` method to position elements to be popped::
536
537 def delete_nth(d, n):
538 d.rotate(-n)
539 d.popleft()
540 d.rotate(n)
541
542To implement :class:`deque` slicing, use a similar approach applying
543:meth:`rotate` to bring a target element to the left side of the deque. Remove
544old entries with :meth:`popleft`, add new entries with :meth:`extend`, and then
545reverse the rotation.
Georg Brandl116aa622007-08-15 14:28:22 +0000546With minor variations on that approach, it is easy to implement Forth style
547stack manipulations such as ``dup``, ``drop``, ``swap``, ``over``, ``pick``,
548``rot``, and ``roll``.
549
Georg Brandl116aa622007-08-15 14:28:22 +0000550
551:class:`defaultdict` objects
552----------------------------
553
Georg Brandl116aa622007-08-15 14:28:22 +0000554.. class:: defaultdict([default_factory[, ...]])
555
556 Returns a new dictionary-like object. :class:`defaultdict` is a subclass of the
Georg Brandl22b34312009-07-26 14:54:51 +0000557 built-in :class:`dict` class. It overrides one method and adds one writable
Georg Brandl116aa622007-08-15 14:28:22 +0000558 instance variable. The remaining functionality is the same as for the
559 :class:`dict` class and is not documented here.
560
561 The first argument provides the initial value for the :attr:`default_factory`
562 attribute; it defaults to ``None``. All remaining arguments are treated the same
563 as if they were passed to the :class:`dict` constructor, including keyword
564 arguments.
565
Georg Brandl116aa622007-08-15 14:28:22 +0000566
Benjamin Petersone41251e2008-04-25 01:59:09 +0000567 :class:`defaultdict` objects support the following method in addition to the
568 standard :class:`dict` operations:
Georg Brandl116aa622007-08-15 14:28:22 +0000569
Benjamin Petersond319ad52010-07-18 14:27:02 +0000570 .. method:: __missing__(key)
Georg Brandl116aa622007-08-15 14:28:22 +0000571
Benjamin Peterson5478b472008-09-17 22:25:09 +0000572 If the :attr:`default_factory` attribute is ``None``, this raises a
Benjamin Petersone41251e2008-04-25 01:59:09 +0000573 :exc:`KeyError` exception with the *key* as argument.
Georg Brandl116aa622007-08-15 14:28:22 +0000574
Benjamin Petersone41251e2008-04-25 01:59:09 +0000575 If :attr:`default_factory` is not ``None``, it is called without arguments
576 to provide a default value for the given *key*, this value is inserted in
577 the dictionary for the *key*, and returned.
Georg Brandl116aa622007-08-15 14:28:22 +0000578
Benjamin Petersone41251e2008-04-25 01:59:09 +0000579 If calling :attr:`default_factory` raises an exception this exception is
580 propagated unchanged.
Georg Brandl116aa622007-08-15 14:28:22 +0000581
Benjamin Petersone41251e2008-04-25 01:59:09 +0000582 This method is called by the :meth:`__getitem__` method of the
583 :class:`dict` class when the requested key is not found; whatever it
584 returns or raises is then returned or raised by :meth:`__getitem__`.
Georg Brandl116aa622007-08-15 14:28:22 +0000585
586
Benjamin Petersone41251e2008-04-25 01:59:09 +0000587 :class:`defaultdict` objects support the following instance variable:
Georg Brandl116aa622007-08-15 14:28:22 +0000588
Benjamin Petersone41251e2008-04-25 01:59:09 +0000589
Benjamin Petersond319ad52010-07-18 14:27:02 +0000590 .. attribute:: default_factory
Benjamin Petersone41251e2008-04-25 01:59:09 +0000591
592 This attribute is used by the :meth:`__missing__` method; it is
593 initialized from the first argument to the constructor, if present, or to
594 ``None``, if absent.
Georg Brandl116aa622007-08-15 14:28:22 +0000595
596
Georg Brandl116aa622007-08-15 14:28:22 +0000597:class:`defaultdict` Examples
598^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
599
600Using :class:`list` as the :attr:`default_factory`, it is easy to group a
Christian Heimesfe337bf2008-03-23 21:54:12 +0000601sequence of key-value pairs into a dictionary of lists:
Georg Brandl116aa622007-08-15 14:28:22 +0000602
603 >>> s = [('yellow', 1), ('blue', 2), ('yellow', 3), ('blue', 4), ('red', 1)]
604 >>> d = defaultdict(list)
605 >>> for k, v in s:
606 ... d[k].append(v)
607 ...
Ezio Melottic53a8942009-09-12 01:52:05 +0000608 >>> list(d.items())
Georg Brandl116aa622007-08-15 14:28:22 +0000609 [('blue', [2, 4]), ('red', [1]), ('yellow', [1, 3])]
610
611When each key is encountered for the first time, it is not already in the
612mapping; so an entry is automatically created using the :attr:`default_factory`
613function which returns an empty :class:`list`. The :meth:`list.append`
614operation then attaches the value to the new list. When keys are encountered
615again, the look-up proceeds normally (returning the list for that key) and the
616:meth:`list.append` operation adds another value to the list. This technique is
Christian Heimesfe337bf2008-03-23 21:54:12 +0000617simpler and faster than an equivalent technique using :meth:`dict.setdefault`:
Georg Brandl116aa622007-08-15 14:28:22 +0000618
619 >>> d = {}
620 >>> for k, v in s:
621 ... d.setdefault(k, []).append(v)
622 ...
Ezio Melottic53a8942009-09-12 01:52:05 +0000623 >>> list(d.items())
Georg Brandl116aa622007-08-15 14:28:22 +0000624 [('blue', [2, 4]), ('red', [1]), ('yellow', [1, 3])]
625
626Setting the :attr:`default_factory` to :class:`int` makes the
627:class:`defaultdict` useful for counting (like a bag or multiset in other
Christian Heimesfe337bf2008-03-23 21:54:12 +0000628languages):
Georg Brandl116aa622007-08-15 14:28:22 +0000629
630 >>> s = 'mississippi'
631 >>> d = defaultdict(int)
632 >>> for k in s:
633 ... d[k] += 1
634 ...
Ezio Melottic53a8942009-09-12 01:52:05 +0000635 >>> list(d.items())
Georg Brandl116aa622007-08-15 14:28:22 +0000636 [('i', 4), ('p', 2), ('s', 4), ('m', 1)]
637
638When a letter is first encountered, it is missing from the mapping, so the
639:attr:`default_factory` function calls :func:`int` to supply a default count of
640zero. The increment operation then builds up the count for each letter.
641
642The function :func:`int` which always returns zero is just a special case of
643constant functions. A faster and more flexible way to create constant functions
644is to use a lambda function which can supply any constant value (not just
Christian Heimesfe337bf2008-03-23 21:54:12 +0000645zero):
Georg Brandl116aa622007-08-15 14:28:22 +0000646
647 >>> def constant_factory(value):
648 ... return lambda: value
649 >>> d = defaultdict(constant_factory('<missing>'))
650 >>> d.update(name='John', action='ran')
651 >>> '%(name)s %(action)s to %(object)s' % d
652 'John ran to <missing>'
653
654Setting the :attr:`default_factory` to :class:`set` makes the
Christian Heimesfe337bf2008-03-23 21:54:12 +0000655:class:`defaultdict` useful for building a dictionary of sets:
Georg Brandl116aa622007-08-15 14:28:22 +0000656
657 >>> s = [('red', 1), ('blue', 2), ('red', 3), ('blue', 4), ('red', 1), ('blue', 4)]
658 >>> d = defaultdict(set)
659 >>> for k, v in s:
660 ... d[k].add(v)
661 ...
Ezio Melottic53a8942009-09-12 01:52:05 +0000662 >>> list(d.items())
Georg Brandl116aa622007-08-15 14:28:22 +0000663 [('blue', set([2, 4])), ('red', set([1, 3]))]
664
665
Guido van Rossum3d392eb2007-11-16 00:35:22 +0000666:func:`namedtuple` Factory Function for Tuples with Named Fields
Christian Heimes790c8232008-01-07 21:14:23 +0000667----------------------------------------------------------------
Georg Brandl116aa622007-08-15 14:28:22 +0000668
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000669Named tuples assign meaning to each position in a tuple and allow for more readable,
670self-documenting code. They can be used wherever regular tuples are used, and
671they add the ability to access fields by name instead of position index.
Georg Brandl116aa622007-08-15 14:28:22 +0000672
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000673.. function:: namedtuple(typename, field_names, verbose=False, rename=False)
Georg Brandl116aa622007-08-15 14:28:22 +0000674
675 Returns a new tuple subclass named *typename*. The new subclass is used to
Christian Heimesc3f30c42008-02-22 16:37:40 +0000676 create tuple-like objects that have fields accessible by attribute lookup as
Georg Brandl116aa622007-08-15 14:28:22 +0000677 well as being indexable and iterable. Instances of the subclass also have a
Benjamin Peterson4469d0c2008-11-30 22:46:23 +0000678 helpful docstring (with typename and field_names) and a helpful :meth:`__repr__`
Georg Brandl116aa622007-08-15 14:28:22 +0000679 method which lists the tuple contents in a ``name=value`` format.
680
Benjamin Peterson4469d0c2008-11-30 22:46:23 +0000681 The *field_names* are a single string with each fieldname separated by whitespace
682 and/or commas, for example ``'x y'`` or ``'x, y'``. Alternatively, *field_names*
Christian Heimes25bb7832008-01-11 16:17:00 +0000683 can be a sequence of strings such as ``['x', 'y']``.
Georg Brandl9afde1c2007-11-01 20:32:30 +0000684
685 Any valid Python identifier may be used for a fieldname except for names
Christian Heimes0449f632007-12-15 01:27:15 +0000686 starting with an underscore. Valid identifiers consist of letters, digits,
687 and underscores but do not start with a digit or underscore and cannot be
Georg Brandlf6945182008-02-01 11:56:49 +0000688 a :mod:`keyword` such as *class*, *for*, *return*, *global*, *pass*,
Georg Brandl9afde1c2007-11-01 20:32:30 +0000689 or *raise*.
Georg Brandl116aa622007-08-15 14:28:22 +0000690
Benjamin Petersona86f2c02009-02-10 02:41:10 +0000691 If *rename* is true, invalid fieldnames are automatically replaced
692 with positional names. For example, ``['abc', 'def', 'ghi', 'abc']`` is
Raymond Hettinger85737b82009-04-02 22:37:59 +0000693 converted to ``['abc', '_1', 'ghi', '_3']``, eliminating the keyword
Benjamin Petersona86f2c02009-02-10 02:41:10 +0000694 ``def`` and the duplicate fieldname ``abc``.
695
Christian Heimes25bb7832008-01-11 16:17:00 +0000696 If *verbose* is true, the class definition is printed just before being built.
Georg Brandl116aa622007-08-15 14:28:22 +0000697
Georg Brandl9afde1c2007-11-01 20:32:30 +0000698 Named tuple instances do not have per-instance dictionaries, so they are
Thomas Wouters8ce81f72007-09-20 18:22:40 +0000699 lightweight and require no more memory than regular tuples.
Georg Brandl116aa622007-08-15 14:28:22 +0000700
Raymond Hettingerb62ad242009-03-02 22:16:43 +0000701 .. versionchanged:: 3.1
Georg Brandl67b21b72010-08-17 15:07:14 +0000702 Added support for *rename*.
Benjamin Petersona86f2c02009-02-10 02:41:10 +0000703
Christian Heimesfe337bf2008-03-23 21:54:12 +0000704
705.. doctest::
706 :options: +NORMALIZE_WHITESPACE
Georg Brandl116aa622007-08-15 14:28:22 +0000707
Raymond Hettinger0ef956f2010-11-21 23:23:29 +0000708 >>> # Basic example
709 >>> Point = namedtuple('Point', 'x y')
710 >>> p = Point(x=10, y=11)
711
712 >>> # Example using the verbose option to print the class definition
Guido van Rossum3d392eb2007-11-16 00:35:22 +0000713 >>> Point = namedtuple('Point', 'x y', verbose=True)
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000714 class Point(tuple):
715 'Point(x, y)'
Christian Heimesfe337bf2008-03-23 21:54:12 +0000716 <BLANKLINE>
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000717 __slots__ = ()
Christian Heimesfe337bf2008-03-23 21:54:12 +0000718 <BLANKLINE>
Christian Heimesfaf2f632008-01-06 16:59:19 +0000719 _fields = ('x', 'y')
Christian Heimesfe337bf2008-03-23 21:54:12 +0000720 <BLANKLINE>
Raymond Hettinger089ba7f2009-05-27 00:38:24 +0000721 def __new__(_cls, x, y):
Raymond Hettinger7b0d3c62010-04-02 18:54:02 +0000722 'Create a new instance of Point(x, y)'
Raymond Hettinger089ba7f2009-05-27 00:38:24 +0000723 return _tuple.__new__(_cls, (x, y))
Christian Heimesfe337bf2008-03-23 21:54:12 +0000724 <BLANKLINE>
Christian Heimesfaf2f632008-01-06 16:59:19 +0000725 @classmethod
Christian Heimesfe337bf2008-03-23 21:54:12 +0000726 def _make(cls, iterable, new=tuple.__new__, len=len):
Christian Heimesfaf2f632008-01-06 16:59:19 +0000727 'Make a new Point object from a sequence or iterable'
Christian Heimesfe337bf2008-03-23 21:54:12 +0000728 result = new(cls, iterable)
Christian Heimesfaf2f632008-01-06 16:59:19 +0000729 if len(result) != 2:
730 raise TypeError('Expected 2 arguments, got %d' % len(result))
731 return result
Christian Heimesfe337bf2008-03-23 21:54:12 +0000732 <BLANKLINE>
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000733 def __repr__(self):
Raymond Hettinger7b0d3c62010-04-02 18:54:02 +0000734 'Return a nicely formatted representation string'
Raymond Hettingerd331ce92010-08-08 01:13:42 +0000735 return self.__class__.__name__ + '(x=%r, y=%r)' % self
Christian Heimesfe337bf2008-03-23 21:54:12 +0000736 <BLANKLINE>
Raymond Hettingera4f52b12009-03-02 22:28:31 +0000737 def _asdict(self):
738 'Return a new OrderedDict which maps field names to their values'
739 return OrderedDict(zip(self._fields, self))
Christian Heimesfe337bf2008-03-23 21:54:12 +0000740 <BLANKLINE>
Raymond Hettinger089ba7f2009-05-27 00:38:24 +0000741 def _replace(_self, **kwds):
Guido van Rossum3d392eb2007-11-16 00:35:22 +0000742 'Return a new Point object replacing specified fields with new values'
Raymond Hettinger089ba7f2009-05-27 00:38:24 +0000743 result = _self._make(map(kwds.pop, ('x', 'y'), _self))
Christian Heimesfaf2f632008-01-06 16:59:19 +0000744 if kwds:
Ezio Melotti8f7649e2009-09-13 04:48:45 +0000745 raise ValueError('Got unexpected field names: %r' % list(kwds.keys()))
Christian Heimesfaf2f632008-01-06 16:59:19 +0000746 return result
Georg Brandl48310cd2009-01-03 21:18:54 +0000747 <BLANKLINE>
748 def __getnewargs__(self):
Raymond Hettinger7b0d3c62010-04-02 18:54:02 +0000749 'Return self as a plain tuple. Used by copy and pickle.'
Benjamin Peterson41181742008-07-02 20:22:54 +0000750 return tuple(self)
Christian Heimesfe337bf2008-03-23 21:54:12 +0000751 <BLANKLINE>
Raymond Hettinger7b0d3c62010-04-02 18:54:02 +0000752 x = _property(_itemgetter(0), doc='Alias for field number 0')
753 y = _property(_itemgetter(1), doc='Alias for field number 1')
Georg Brandl116aa622007-08-15 14:28:22 +0000754
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000755 >>> p = Point(11, y=22) # instantiate with positional or keyword arguments
Christian Heimes99170a52007-12-19 02:07:34 +0000756 >>> p[0] + p[1] # indexable like the plain tuple (11, 22)
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000757 33
758 >>> x, y = p # unpack like a regular tuple
759 >>> x, y
760 (11, 22)
Christian Heimesc3f30c42008-02-22 16:37:40 +0000761 >>> p.x + p.y # fields also accessible by name
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000762 33
763 >>> p # readable __repr__ with a name=value style
764 Point(x=11, y=22)
Georg Brandl116aa622007-08-15 14:28:22 +0000765
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000766Named tuples are especially useful for assigning field names to result tuples returned
767by the :mod:`csv` or :mod:`sqlite3` modules::
768
Guido van Rossum3d392eb2007-11-16 00:35:22 +0000769 EmployeeRecord = namedtuple('EmployeeRecord', 'name, age, title, department, paygrade')
Georg Brandl9afde1c2007-11-01 20:32:30 +0000770
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000771 import csv
Christian Heimesfaf2f632008-01-06 16:59:19 +0000772 for emp in map(EmployeeRecord._make, csv.reader(open("employees.csv", "rb"))):
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000773 print(emp.name, emp.title)
774
Georg Brandl9afde1c2007-11-01 20:32:30 +0000775 import sqlite3
776 conn = sqlite3.connect('/companydata')
777 cursor = conn.cursor()
778 cursor.execute('SELECT name, age, title, department, paygrade FROM employees')
Christian Heimesfaf2f632008-01-06 16:59:19 +0000779 for emp in map(EmployeeRecord._make, cursor.fetchall()):
Christian Heimes00412232008-01-10 16:02:19 +0000780 print(emp.name, emp.title)
Georg Brandl9afde1c2007-11-01 20:32:30 +0000781
Christian Heimes99170a52007-12-19 02:07:34 +0000782In addition to the methods inherited from tuples, named tuples support
Christian Heimes2380ac72008-01-09 00:17:24 +0000783three additional methods and one attribute. To prevent conflicts with
784field names, the method and attribute names start with an underscore.
Christian Heimes99170a52007-12-19 02:07:34 +0000785
Benjamin Peterson0b9fb802010-07-18 14:23:36 +0000786.. classmethod:: somenamedtuple._make(iterable)
Christian Heimes99170a52007-12-19 02:07:34 +0000787
Christian Heimesfaf2f632008-01-06 16:59:19 +0000788 Class method that makes a new instance from an existing sequence or iterable.
Christian Heimes99170a52007-12-19 02:07:34 +0000789
Christian Heimesfe337bf2008-03-23 21:54:12 +0000790.. doctest::
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000791
Christian Heimesfaf2f632008-01-06 16:59:19 +0000792 >>> t = [11, 22]
793 >>> Point._make(t)
794 Point(x=11, y=22)
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000795
Christian Heimes790c8232008-01-07 21:14:23 +0000796.. method:: somenamedtuple._asdict()
Georg Brandl9afde1c2007-11-01 20:32:30 +0000797
Raymond Hettingera4f52b12009-03-02 22:28:31 +0000798 Return a new :class:`OrderedDict` which maps field names to their corresponding
799 values::
Georg Brandl9afde1c2007-11-01 20:32:30 +0000800
Christian Heimes0449f632007-12-15 01:27:15 +0000801 >>> p._asdict()
Raymond Hettingera4f52b12009-03-02 22:28:31 +0000802 OrderedDict([('x', 11), ('y', 22)])
803
Raymond Hettingera88e4da2009-03-03 05:12:27 +0000804 .. versionchanged:: 3.1
Raymond Hettingera4f52b12009-03-02 22:28:31 +0000805 Returns an :class:`OrderedDict` instead of a regular :class:`dict`.
Christian Heimesfe337bf2008-03-23 21:54:12 +0000806
Christian Heimes790c8232008-01-07 21:14:23 +0000807.. method:: somenamedtuple._replace(kwargs)
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000808
Christian Heimesfe337bf2008-03-23 21:54:12 +0000809 Return a new instance of the named tuple replacing specified fields with new
810 values:
Thomas Wouters8ce81f72007-09-20 18:22:40 +0000811
812::
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000813
814 >>> p = Point(x=11, y=22)
Christian Heimes0449f632007-12-15 01:27:15 +0000815 >>> p._replace(x=33)
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000816 Point(x=33, y=22)
817
Guido van Rossum3d392eb2007-11-16 00:35:22 +0000818 >>> for partnum, record in inventory.items():
Christian Heimes454f37b2008-01-10 00:10:02 +0000819 ... inventory[partnum] = record._replace(price=newprices[partnum], timestamp=time.now())
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000820
Christian Heimes790c8232008-01-07 21:14:23 +0000821.. attribute:: somenamedtuple._fields
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000822
Christian Heimes2380ac72008-01-09 00:17:24 +0000823 Tuple of strings listing the field names. Useful for introspection
Georg Brandl9afde1c2007-11-01 20:32:30 +0000824 and for creating new named tuple types from existing named tuples.
Thomas Wouters8ce81f72007-09-20 18:22:40 +0000825
Christian Heimesfe337bf2008-03-23 21:54:12 +0000826.. doctest::
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000827
Christian Heimes0449f632007-12-15 01:27:15 +0000828 >>> p._fields # view the field names
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000829 ('x', 'y')
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000830
Guido van Rossum3d392eb2007-11-16 00:35:22 +0000831 >>> Color = namedtuple('Color', 'red green blue')
Christian Heimes0449f632007-12-15 01:27:15 +0000832 >>> Pixel = namedtuple('Pixel', Point._fields + Color._fields)
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000833 >>> Pixel(11, 22, 128, 255, 0)
Christian Heimes454f37b2008-01-10 00:10:02 +0000834 Pixel(x=11, y=22, red=128, green=255, blue=0)
Georg Brandl116aa622007-08-15 14:28:22 +0000835
Christian Heimes0449f632007-12-15 01:27:15 +0000836To retrieve a field whose name is stored in a string, use the :func:`getattr`
Christian Heimesfe337bf2008-03-23 21:54:12 +0000837function:
Christian Heimes0449f632007-12-15 01:27:15 +0000838
839 >>> getattr(p, 'x')
840 11
841
Raymond Hettinger651453a2009-02-11 00:20:02 +0000842To convert a dictionary to a named tuple, use the double-star-operator
843(as described in :ref:`tut-unpacking-arguments`):
Christian Heimes99170a52007-12-19 02:07:34 +0000844
845 >>> d = {'x': 11, 'y': 22}
846 >>> Point(**d)
847 Point(x=11, y=22)
848
Guido van Rossum3d392eb2007-11-16 00:35:22 +0000849Since a named tuple is a regular Python class, it is easy to add or change
Christian Heimes043d6f62008-01-07 17:19:16 +0000850functionality with a subclass. Here is how to add a calculated field and
Christian Heimesfe337bf2008-03-23 21:54:12 +0000851a fixed-width print format:
Guido van Rossum3d392eb2007-11-16 00:35:22 +0000852
Christian Heimes043d6f62008-01-07 17:19:16 +0000853 >>> class Point(namedtuple('Point', 'x y')):
Christian Heimes25bb7832008-01-11 16:17:00 +0000854 ... __slots__ = ()
Christian Heimes454f37b2008-01-10 00:10:02 +0000855 ... @property
856 ... def hypot(self):
857 ... return (self.x ** 2 + self.y ** 2) ** 0.5
858 ... def __str__(self):
Christian Heimes25bb7832008-01-11 16:17:00 +0000859 ... 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 +0000860
Georg Brandl0df79792008-10-04 18:33:26 +0000861 >>> for p in Point(3, 4), Point(14, 5/7):
Christian Heimes00412232008-01-10 16:02:19 +0000862 ... print(p)
Christian Heimes25bb7832008-01-11 16:17:00 +0000863 Point: x= 3.000 y= 4.000 hypot= 5.000
864 Point: x=14.000 y= 0.714 hypot=14.018
Christian Heimes043d6f62008-01-07 17:19:16 +0000865
Georg Brandlaf5c2382009-12-28 08:02:38 +0000866The subclass shown above sets ``__slots__`` to an empty tuple. This helps
Christian Heimes679db4a2008-01-18 09:56:22 +0000867keep memory requirements low by preventing the creation of instance dictionaries.
868
Christian Heimes2380ac72008-01-09 00:17:24 +0000869
870Subclassing is not useful for adding new, stored fields. Instead, simply
Christian Heimesfe337bf2008-03-23 21:54:12 +0000871create a new named tuple type from the :attr:`_fields` attribute:
Christian Heimes2380ac72008-01-09 00:17:24 +0000872
Christian Heimes25bb7832008-01-11 16:17:00 +0000873 >>> Point3D = namedtuple('Point3D', Point._fields + ('z',))
Christian Heimes2380ac72008-01-09 00:17:24 +0000874
875Default values can be implemented by using :meth:`_replace` to
Christian Heimesfe337bf2008-03-23 21:54:12 +0000876customize a prototype instance:
Guido van Rossum3d392eb2007-11-16 00:35:22 +0000877
878 >>> Account = namedtuple('Account', 'owner balance transaction_count')
Christian Heimes587c2bf2008-01-19 16:21:02 +0000879 >>> default_account = Account('<owner name>', 0.0, 0)
880 >>> johns_account = default_account._replace(owner='John')
Guido van Rossum3d392eb2007-11-16 00:35:22 +0000881
Christian Heimese4ca8152008-05-08 17:18:53 +0000882Enumerated constants can be implemented with named tuples, but it is simpler
883and more efficient to use a simple class declaration:
884
885 >>> Status = namedtuple('Status', 'open pending closed')._make(range(3))
886 >>> Status.open, Status.pending, Status.closed
887 (0, 1, 2)
888 >>> class Status:
889 ... open, pending, closed = range(3)
890
Raymond Hettinger651453a2009-02-11 00:20:02 +0000891.. seealso::
Thomas Wouters47b49bf2007-08-30 22:15:33 +0000892
Raymond Hettinger651453a2009-02-11 00:20:02 +0000893 `Named tuple recipe <http://code.activestate.com/recipes/500261/>`_
894 adapted for Python 2.4.
Raymond Hettingere4c96ad2008-02-06 01:23:58 +0000895
896
Raymond Hettinger2d32f632009-03-02 21:24:57 +0000897:class:`OrderedDict` objects
898----------------------------
899
900Ordered dictionaries are just like regular dictionaries but they remember the
901order that items were inserted. When iterating over an ordered dictionary,
902the items are returned in the order their keys were first added.
903
904.. class:: OrderedDict([items])
905
906 Return an instance of a dict subclass, supporting the usual :class:`dict`
907 methods. An *OrderedDict* is a dict that remembers the order that keys
908 were first inserted. If a new entry overwrites an existing entry, the
909 original insertion position is left unchanged. Deleting an entry and
910 reinserting it will move it to the end.
911
Benjamin Petersond45bf582009-03-02 21:44:54 +0000912 .. versionadded:: 3.1
Raymond Hettinger2d32f632009-03-02 21:24:57 +0000913
Benjamin Petersond319ad52010-07-18 14:27:02 +0000914 .. method:: popitem(last=True)
Raymond Hettingerdc879f02009-03-19 20:30:56 +0000915
Benjamin Petersond319ad52010-07-18 14:27:02 +0000916 The :meth:`popitem` method for ordered dictionaries returns and removes a
917 (key, value) pair. The pairs are returned in LIFO order if *last* is true
918 or FIFO order if false.
Raymond Hettinger2d32f632009-03-02 21:24:57 +0000919
Raymond Hettingerf45abc92010-09-06 21:26:09 +0000920 .. method:: move_to_end(key, last=True)
921
922 Move an existing *key* to either end of an ordered dictionary. The item
923 is moved to the right end if *last* is true (the default) or to the
924 beginning if *last* is false. Raises :exc:`KeyError` if the *key* does
925 not exist::
926
927 >>> d = OrderedDict.fromkeys('abcde')
928 >>> d.move_to_end('b')
929 >>> ''.join(d.keys)
930 'acdeb'
Éric Araujo1cb25aa2010-11-06 07:03:07 +0000931 >>> d.move_to_end('b', last=False)
Raymond Hettingerf45abc92010-09-06 21:26:09 +0000932 >>> ''.join(d.keys)
933 'bacde'
934
935 .. versionadded:: 3.2
936
Raymond Hettingere9091502009-05-19 17:40:07 +0000937In addition to the usual mapping methods, ordered dictionaries also support
938reverse iteration using :func:`reversed`.
939
Raymond Hettinger2d32f632009-03-02 21:24:57 +0000940Equality tests between :class:`OrderedDict` objects are order-sensitive
941and are implemented as ``list(od1.items())==list(od2.items())``.
942Equality tests between :class:`OrderedDict` objects and other
943:class:`Mapping` objects are order-insensitive like regular dictionaries.
944This allows :class:`OrderedDict` objects to be substituted anywhere a
945regular dictionary is used.
946
Raymond Hettinger36180782009-04-09 22:34:23 +0000947The :class:`OrderedDict` constructor and :meth:`update` method both accept
948keyword arguments, but their order is lost because Python's function call
949semantics pass-in keyword arguments using a regular unordered dictionary.
950
Raymond Hettingerdc879f02009-03-19 20:30:56 +0000951.. seealso::
952
953 `Equivalent OrderedDict recipe <http://code.activestate.com/recipes/576693/>`_
954 that runs on Python 2.4 or later.
955
Raymond Hettinger0e312012009-11-10 18:35:46 +0000956Since an ordered dictionary remembers its insertion order, it can be used
957in conjuction with sorting to make a sorted dictionary::
958
959 >>> # regular unsorted dictionary
960 >>> d = {'banana': 3, 'apple':4, 'pear': 1, 'orange': 2}
961
962 >>> # dictionary sorted by key
963 >>> OrderedDict(sorted(d.items(), key=lambda t: t[0]))
964 OrderedDict([('apple', 4), ('banana', 3), ('orange', 2), ('pear', 1)])
965
966 >>> # dictionary sorted by value
967 >>> OrderedDict(sorted(d.items(), key=lambda t: t[1]))
968 OrderedDict([('pear', 1), ('orange', 2), ('banana', 3), ('apple', 4)])
969
970 >>> # dictionary sorted by length of the key string
971 >>> OrderedDict(sorted(d.items(), key=lambda t: len(t[0])))
972 OrderedDict([('pear', 1), ('apple', 4), ('orange', 2), ('banana', 3)])
973
974The new sorted dictionaries maintain their sort order when entries
975are deleted. But when new keys are added, the keys are appended
976to the end and the sort is not maintained.
977
Raymond Hettinger4821ef82010-07-31 10:14:41 +0000978It is also straight-forward to create an ordered dictionary variant
979that the remembers the order the keys were *last* inserted.
980If a new entry overwrites an existing entry, the
981original insertion position is changed and moved to the end::
982
983 class LastUpdatedOrderedDict(OrderedDict):
Georg Brandl77570e22010-12-18 16:21:58 +0000984 'Store items in the order the keys were last added'
Raymond Hettinger4821ef82010-07-31 10:14:41 +0000985 def __setitem__(self, key, value):
986 if key in self:
987 del self[key]
988 OrderedDict.__setitem__(self, key, value)
989
Raymond Hettingere4c96ad2008-02-06 01:23:58 +0000990
991:class:`UserDict` objects
Mark Summerfield8f2d0062008-02-06 13:30:44 +0000992-------------------------
Raymond Hettingere4c96ad2008-02-06 01:23:58 +0000993
Georg Brandl48310cd2009-01-03 21:18:54 +0000994The class, :class:`UserDict` acts as a wrapper around dictionary objects.
995The need for this class has been partially supplanted by the ability to
Raymond Hettingere4c96ad2008-02-06 01:23:58 +0000996subclass directly from :class:`dict`; however, this class can be easier
997to work with because the underlying dictionary is accessible as an
998attribute.
999
1000.. class:: UserDict([initialdata])
1001
1002 Class that simulates a dictionary. The instance's contents are kept in a
1003 regular dictionary, which is accessible via the :attr:`data` attribute of
1004 :class:`UserDict` instances. If *initialdata* is provided, :attr:`data` is
1005 initialized with its contents; note that a reference to *initialdata* will not
1006 be kept, allowing it be used for other purposes.
1007
Benjamin Petersond319ad52010-07-18 14:27:02 +00001008 In addition to supporting the methods and operations of mappings,
1009 :class:`UserDict` instances provide the following attribute:
Raymond Hettingere4c96ad2008-02-06 01:23:58 +00001010
Benjamin Petersond319ad52010-07-18 14:27:02 +00001011 .. attribute:: data
Raymond Hettingere4c96ad2008-02-06 01:23:58 +00001012
Benjamin Petersond319ad52010-07-18 14:27:02 +00001013 A real dictionary used to store the contents of the :class:`UserDict`
1014 class.
Raymond Hettinger53dbe392008-02-12 20:03:09 +00001015
1016
1017
1018:class:`UserList` objects
1019-------------------------
1020
1021This class acts as a wrapper around list objects. It is a useful base class
Georg Brandl48310cd2009-01-03 21:18:54 +00001022for your own list-like classes which can inherit from them and override
Raymond Hettinger53dbe392008-02-12 20:03:09 +00001023existing methods or add new ones. In this way, one can add new behaviors to
1024lists.
1025
Georg Brandl48310cd2009-01-03 21:18:54 +00001026The need for this class has been partially supplanted by the ability to
Raymond Hettinger53dbe392008-02-12 20:03:09 +00001027subclass directly from :class:`list`; however, this class can be easier
1028to work with because the underlying list is accessible as an attribute.
1029
1030.. class:: UserList([list])
1031
1032 Class that simulates a list. The instance's contents are kept in a regular
1033 list, which is accessible via the :attr:`data` attribute of :class:`UserList`
1034 instances. The instance's contents are initially set to a copy of *list*,
1035 defaulting to the empty list ``[]``. *list* can be any iterable, for
1036 example a real Python list or a :class:`UserList` object.
1037
Benjamin Petersond319ad52010-07-18 14:27:02 +00001038 In addition to supporting the methods and operations of mutable sequences,
1039 :class:`UserList` instances provide the following attribute:
Raymond Hettinger53dbe392008-02-12 20:03:09 +00001040
Benjamin Petersond319ad52010-07-18 14:27:02 +00001041 .. attribute:: data
Raymond Hettinger53dbe392008-02-12 20:03:09 +00001042
Benjamin Petersond319ad52010-07-18 14:27:02 +00001043 A real :class:`list` object used to store the contents of the
1044 :class:`UserList` class.
Raymond Hettinger53dbe392008-02-12 20:03:09 +00001045
1046**Subclassing requirements:** Subclasses of :class:`UserList` are expect to
1047offer a constructor which can be called with either no arguments or one
1048argument. List operations which return a new sequence attempt to create an
1049instance of the actual implementation class. To do so, it assumes that the
1050constructor can be called with a single parameter, which is a sequence object
1051used as a data source.
1052
1053If a derived class does not wish to comply with this requirement, all of the
1054special methods supported by this class will need to be overridden; please
1055consult the sources for information about the methods which need to be provided
1056in that case.
Raymond Hettingerb3a65f82008-02-21 22:11:37 +00001057
1058:class:`UserString` objects
Christian Heimesc3f30c42008-02-22 16:37:40 +00001059---------------------------
Raymond Hettingerb3a65f82008-02-21 22:11:37 +00001060
Georg Brandl48310cd2009-01-03 21:18:54 +00001061The class, :class:`UserString` acts as a wrapper around string objects.
1062The need for this class has been partially supplanted by the ability to
Raymond Hettingerb3a65f82008-02-21 22:11:37 +00001063subclass directly from :class:`str`; however, this class can be easier
1064to work with because the underlying string is accessible as an
1065attribute.
1066
1067.. class:: UserString([sequence])
1068
1069 Class that simulates a string or a Unicode string object. The instance's
Georg Brandl48310cd2009-01-03 21:18:54 +00001070 content is kept in a regular string object, which is accessible via the
1071 :attr:`data` attribute of :class:`UserString` instances. The instance's
Raymond Hettingerb3a65f82008-02-21 22:11:37 +00001072 contents are initially set to a copy of *sequence*. The *sequence* can
1073 be an instance of :class:`bytes`, :class:`str`, :class:`UserString` (or a
1074 subclass) or an arbitrary sequence which can be converted into a string using
1075 the built-in :func:`str` function.