blob: 4cf3a48b03f8b6b3ee11c7b657a40596c4cd1e0f [file] [log] [blame]
Georg Brandl116aa622007-08-15 14:28:22 +00001
Raymond Hettinger53dbe392008-02-12 20:03:09 +00002:mod:`collections` --- Container datatypes
3==========================================
Georg Brandl116aa622007-08-15 14:28:22 +00004
5.. module:: collections
Raymond Hettinger53dbe392008-02-12 20:03:09 +00006 :synopsis: Container datatypes
Georg Brandl116aa622007-08-15 14:28:22 +00007.. moduleauthor:: Raymond Hettinger <python@rcn.com>
8.. sectionauthor:: Raymond Hettinger <python@rcn.com>
9
Christian Heimesfe337bf2008-03-23 21:54:12 +000010.. testsetup:: *
11
12 from collections import *
13 import itertools
14 __name__ = '<doctest>'
Georg Brandl116aa622007-08-15 14:28:22 +000015
Georg Brandl116aa622007-08-15 14:28:22 +000016This module implements high-performance container datatypes. Currently,
17there are two datatypes, :class:`deque` and :class:`defaultdict`, and
Mark Summerfield71316b02008-02-14 16:28:00 +000018one datatype factory function, :func:`namedtuple`. This module also
19provides the :class:`UserDict` and :class:`UserList` classes which may
20be useful when inheriting directly from :class:`dict` or
21:class:`list` isn't convenient.
Christian Heimes0bd4e112008-02-12 22:59:25 +000022
Raymond Hettingerebcee3f2008-02-06 19:54:00 +000023The specialized containers provided in this module provide alternatives
Christian Heimesfe337bf2008-03-23 21:54:12 +000024to Python's general purpose built-in containers, :class:`dict`,
Raymond Hettingerebcee3f2008-02-06 19:54:00 +000025:class:`list`, :class:`set`, and :class:`tuple`.
Georg Brandl116aa622007-08-15 14:28:22 +000026
Mark Summerfield08898b42007-09-05 08:43:04 +000027In addition to containers, the collections module provides some ABCs
Christian Heimesfe337bf2008-03-23 21:54:12 +000028(abstract base classes) that can be used to test whether a class
Raymond Hettingerebcee3f2008-02-06 19:54:00 +000029provides a particular interface, for example, is it hashable or
Mark Summerfield71316b02008-02-14 16:28:00 +000030a mapping, and some of them can also be used as mixin classes.
Raymond Hettingerebcee3f2008-02-06 19:54:00 +000031
32ABCs - abstract base classes
33----------------------------
34
35The collections module offers the following ABCs:
Mark Summerfield08898b42007-09-05 08:43:04 +000036
Georg Brandl86b2fb92008-07-16 03:43:04 +000037========================= ===================== ====================== ====================================================
38ABC Inherits Abstract Methods Mixin Methods
39========================= ===================== ====================== ====================================================
40:class:`Container` ``__contains__``
41:class:`Hashable` ``__hash__``
42:class:`Iterable` ``__iter__``
43:class:`Iterator` :class:`Iterable` ``__next__`` ``__iter__``
Georg Brandla1c6a1c2009-01-03 21:26:05 +000044:class:`Sized` ``__len__``
Georg Brandl86b2fb92008-07-16 03:43:04 +000045:class:`Callable` ``__call__``
Georg Brandl48310cd2009-01-03 21:18:54 +000046
Georg Brandl86b2fb92008-07-16 03:43:04 +000047:class:`Sequence` :class:`Sized`, ``__getitem__`` ``__contains__``. ``__iter__``, ``__reversed__``.
48 :class:`Iterable`, and ``__len__`` ``index``, and ``count``
Georg Brandl48310cd2009-01-03 21:18:54 +000049 :class:`Container`
50
Benjamin Peterson4469d0c2008-11-30 22:46:23 +000051:class:`MutableSequence` :class:`Sequence` ``__getitem__`` Inherited Sequence methods and
Georg Brandl86b2fb92008-07-16 03:43:04 +000052 ``__delitem__``, ``append``, ``reverse``, ``extend``, ``pop``,
53 ``insert``, ``remove``, and ``__iadd__``
54 and ``__len__``
Georg Brandl48310cd2009-01-03 21:18:54 +000055
Georg Brandl86b2fb92008-07-16 03:43:04 +000056:class:`Set` :class:`Sized`, ``__len__``, ``__le__``, ``__lt__``, ``__eq__``, ``__ne__``,
57 :class:`Iterable`, ``__iter__``, and ``__gt__``, ``__ge__``, ``__and__``, ``__or__``
58 :class:`Container` ``__contains__`` ``__sub__``, ``__xor__``, and ``isdisjoint``
Georg Brandl48310cd2009-01-03 21:18:54 +000059
Georg Brandl86b2fb92008-07-16 03:43:04 +000060:class:`MutableSet` :class:`Set` ``add`` and Inherited Set methods and
61 ``discard`` ``clear``, ``pop``, ``remove``, ``__ior__``,
62 ``__iand__``, ``__ixor__``, and ``__isub__``
Georg Brandl48310cd2009-01-03 21:18:54 +000063
Georg Brandl86b2fb92008-07-16 03:43:04 +000064:class:`Mapping` :class:`Sized`, ``__getitem__``, ``__contains__``, ``keys``, ``items``, ``values``,
65 :class:`Iterable`, ``__len__``. and ``get``, ``__eq__``, and ``__ne__``
66 :class:`Container` ``__iter__``
Georg Brandl48310cd2009-01-03 21:18:54 +000067
Georg Brandl86b2fb92008-07-16 03:43:04 +000068:class:`MutableMapping` :class:`Mapping` ``__getitem__`` Inherited Mapping methods and
69 ``__setitem__``, ``pop``, ``popitem``, ``clear``, ``update``,
70 ``__delitem__``, and ``setdefault``
Georg Brandla1c6a1c2009-01-03 21:26:05 +000071 ``__iter__``, and
Georg Brandl86b2fb92008-07-16 03:43:04 +000072 ``__len__``
Georg Brandl48310cd2009-01-03 21:18:54 +000073
Georg Brandl86b2fb92008-07-16 03:43:04 +000074:class:`MappingView` :class:`Sized` ``__len__``
75:class:`KeysView` :class:`MappingView`, ``__contains__``,
76 :class:`Set` ``__iter__``
77:class:`ItemsView` :class:`MappingView`, ``__contains__``,
78 :class:`Set` ``__iter__``
79:class:`ValuesView` :class:`MappingView` ``__contains__``, ``__iter__``
80========================= ===================== ====================== ====================================================
Mark Summerfield08898b42007-09-05 08:43:04 +000081
Mark Summerfield08898b42007-09-05 08:43:04 +000082These ABCs allow us to ask classes or instances if they provide
83particular functionality, for example::
84
Mark Summerfield08898b42007-09-05 08:43:04 +000085 size = None
Raymond Hettingerebcee3f2008-02-06 19:54:00 +000086 if isinstance(myvar, collections.Sized):
Georg Brandla1c6a1c2009-01-03 21:26:05 +000087 size = len(myvar)
Mark Summerfield08898b42007-09-05 08:43:04 +000088
Raymond Hettingerebcee3f2008-02-06 19:54:00 +000089Several of the ABCs are also useful as mixins that make it easier to develop
90classes supporting container APIs. For example, to write a class supporting
91the full :class:`Set` API, it only necessary to supply the three underlying
92abstract methods: :meth:`__contains__`, :meth:`__iter__`, and :meth:`__len__`.
93The ABC supplies the remaining methods such as :meth:`__and__` and
94:meth:`isdisjoint` ::
95
96 class ListBasedSet(collections.Set):
Raymond Hettingerc1b6a4a2008-02-08 23:46:23 +000097 ''' Alternate set implementation favoring space over speed
98 and not requiring the set elements to be hashable. '''
Raymond Hettingerebcee3f2008-02-06 19:54:00 +000099 def __init__(self, iterable):
Raymond Hettingerc1b6a4a2008-02-08 23:46:23 +0000100 self.elements = lst = []
101 for value in iterable:
102 if value not in lst:
103 lst.append(value)
Raymond Hettingerebcee3f2008-02-06 19:54:00 +0000104 def __iter__(self):
105 return iter(self.elements)
106 def __contains__(self, value):
107 return value in self.elements
108 def __len__(self):
109 return len(self.elements)
110
111 s1 = ListBasedSet('abcdef')
112 s2 = ListBasedSet('defghi')
113 overlap = s1 & s2 # The __and__() method is supported automatically
114
Raymond Hettinger7aebb642008-02-09 03:25:08 +0000115Notes on using :class:`Set` and :class:`MutableSet` as a mixin:
116
Christian Heimesfe337bf2008-03-23 21:54:12 +0000117(1)
Raymond Hettinger7aebb642008-02-09 03:25:08 +0000118 Since some set operations create new sets, the default mixin methods need
Christian Heimesfe337bf2008-03-23 21:54:12 +0000119 a way to create new instances from an iterable. The class constructor is
120 assumed to have a signature in the form ``ClassName(iterable)``.
Benjamin Peterson2b7411d2008-05-26 17:36:47 +0000121 That assumption is factored-out to an internal classmethod called
Raymond Hettinger7aebb642008-02-09 03:25:08 +0000122 :meth:`_from_iterable` which calls ``cls(iterable)`` to produce a new set.
123 If the :class:`Set` mixin is being used in a class with a different
Christian Heimesfe337bf2008-03-23 21:54:12 +0000124 constructor signature, you will need to override :meth:`from_iterable`
125 with a classmethod that can construct new instances from
Raymond Hettinger7aebb642008-02-09 03:25:08 +0000126 an iterable argument.
127
128(2)
129 To override the comparisons (presumably for speed, as the
130 semantics are fixed), redefine :meth:`__le__` and
131 then the other operations will automatically follow suit.
Raymond Hettingerebcee3f2008-02-06 19:54:00 +0000132
Raymond Hettinger0dbdab22008-02-09 03:48:16 +0000133(3)
134 The :class:`Set` mixin provides a :meth:`_hash` method to compute a hash value
135 for the set; however, :meth:`__hash__` is not defined because not all sets
136 are hashable or immutable. To add set hashabilty using mixins,
137 inherit from both :meth:`Set` and :meth:`Hashable`, then define
138 ``__hash__ = Set._hash``.
139
Mark Summerfield08898b42007-09-05 08:43:04 +0000140(For more about ABCs, see the :mod:`abc` module and :pep:`3119`.)
141
142
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000143:class:`Counter` objects
144------------------------
145
146A counter tool is provided to support convenient and rapid tallies.
147For example::
148
Raymond Hettinger670eaec2009-01-21 23:14:07 +0000149 # Tally occurrences of words in a list
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000150 >>> cnt = Counter()
Raymond Hettinger670eaec2009-01-21 23:14:07 +0000151 >>> for word in ['red', 'blue', 'red', 'green', 'blue', 'blue']:
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000152 ... cnt[word] += 1
153 >>> cnt
154 Counter({'blue': 3, 'red': 2, 'green': 1})
155
156 # Find the ten most common words in Hamlet
157 >>> import re
158 >>> words = re.findall('\w+', open('hamlet.txt').read().lower())
Raymond Hettinger0bae6622009-01-20 13:00:59 +0000159 >>> Counter(words).most_common(10)
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000160 [('the', 1143), ('and', 966), ('to', 762), ('of', 669), ('i', 631),
161 ('you', 554), ('a', 546), ('my', 514), ('hamlet', 471), ('in', 451)]
162
163.. class:: Counter([iterable-or-mapping])
164
Raymond Hettinger670eaec2009-01-21 23:14:07 +0000165 A :class:`Counter` is a :class:`dict` subclass for counting hashable objects.
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000166 It is an unordered collection where elements are stored as dictionary keys
167 and their counts are stored as dictionary values. Counts are allowed to be
168 any integer value including zero or negative counts. The :class:`Counter`
169 class is similar to bags or multisets in other languages.
170
171 Elements are counted from an *iterable* or initialized from another
172 *mapping* (or counter)::
173
174 >>> c = Counter() # a new, empty counter
175 >>> c = Counter('gallahad') # a new counter from an iterable
176 >>> c = Counter({'red': 4, 'blue': 2}) # a new counter from a mapping
Raymond Hettinger4d2073a2009-01-20 03:41:22 +0000177 >>> c = Counter(spam=8, eggs=1) # a new counter from keyword args
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000178
Raymond Hettinger670eaec2009-01-21 23:14:07 +0000179 Counter objects have a dictionary interface except that they return a zero
180 count for missing items instead of raising a :exc:`KeyError`::
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000181
182 >>> c = Counter(['egg', 'ham'])
183 >>> c['bacon'] # count of a missing element is zero
184 0
185
Raymond Hettinger670eaec2009-01-21 23:14:07 +0000186 Setting a count to zero still leaves an element in the dictionary. Use
187 ``del`` to remove it entirely:
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000188
189 >>> c = Counter(['arthur', 'gwain'])
190 >>> c['arthur'] = 0 # set the count of 'arthur' to zero
191 >>> 'arthur' in c # but 'arthur' is still in the counter
192 True
193 >>> del c['arthur'] # del will completely remove the entry
194
195 .. versionadded:: 2.7
196
197
198 Counter objects support two methods beyond those available for all
199 dictionaries:
200
201 .. method:: elements()
202
Raymond Hettinger670eaec2009-01-21 23:14:07 +0000203 Return an iterator over elements repeating each as many times as its
204 count. Elements are returned in arbitrary order. If an element's count
205 is less than one, :meth:`elements` will ignore it.
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000206
Raymond Hettinger0bae6622009-01-20 13:00:59 +0000207 >>> c = Counter(a=4, b=2, c=0, d=-2)
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000208 >>> list(c.elements())
209 ['a', 'a', 'a', 'a', 'b', 'b']
210
211 .. method:: most_common([n])
212
213 Return a list of the *n* most common elements and their counts from
214 the most common to the least. If *n* is not specified or is ``None``,
215 return a list of all element counts in decreasing order of frequency.
216 Elements with equal counts are ordered arbitrarily::
217
218 >>> Counter('abracadabra').most_common(3)
219 [('a', 5), ('r', 2), ('b', 2)]
220
Raymond Hettinger670eaec2009-01-21 23:14:07 +0000221 The usual dictionary methods are available for :class:`Counter` objects
222 except for two which work differently for counters.
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000223
224 .. method:: fromkeys(iterable)
225
Raymond Hettinger670eaec2009-01-21 23:14:07 +0000226 This class method is not implemented for :class:`Counter` objects.
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000227
228 .. method:: update([iterable-or-mapping])
229
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000230 Elements are counted from an *iterable* or added-in from another
Raymond Hettinger4d2073a2009-01-20 03:41:22 +0000231 *mapping* (or counter). Like :meth:`dict.update` but adds-in counts
Raymond Hettinger670eaec2009-01-21 23:14:07 +0000232 instead of replacing them. Also, the *iterable* is expected to be a
Raymond Hettinger4d2073a2009-01-20 03:41:22 +0000233 sequence of elements, not a sequence of ``(key, value)`` pairs::
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000234
235 >>> c = Counter('which')
236 >>> c.update('witch') # add elements from another iterable
237 >>> d = Counter('watch')
238 >>> c.update(d) # add elements from another counter
239 >>> c['h'] # four 'h' in which, witch, and watch
240 4
241
242Common patterns for working with :class:`Counter` objects::
243
244 sum(c.values()) # total of all counts
245 c.clear() # reset all counts
246 list(c) # list unique elements
247 set(c) # convert to a set
248 dict(c) # convert to a regular dictionary
249 c.items() # convert to a list of (elem, cnt) pairs
250 Counter(dict(list_of_pairs)) # convert from a list of (elem, cnt) pairs
251 c.most_common()[:-n:-1] # n least common elements
Raymond Hettingere0d1b9f2009-01-21 20:36:27 +0000252 c += Counter() # remove zero and negative counts
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000253
Raymond Hettinger4d2073a2009-01-20 03:41:22 +0000254Several multiset mathematical operations are provided for combining
Raymond Hettingere0d1b9f2009-01-21 20:36:27 +0000255:class:`Counter` objects. Multisets are like regular sets but are allowed to
Raymond Hettinger4d2073a2009-01-20 03:41:22 +0000256contain repeated elements (with counts of one or more). Addition and
257subtraction combine counters by adding or subtracting the counts of
258corresponding elements. Intersection and union return the minimum and maximum
Raymond Hettingere0d1b9f2009-01-21 20:36:27 +0000259of corresponding counts. All four multiset operations exclude results with
Raymond Hettinger670eaec2009-01-21 23:14:07 +0000260counts less than one::
Raymond Hettinger4d2073a2009-01-20 03:41:22 +0000261
Raymond Hettingere0d1b9f2009-01-21 20:36:27 +0000262 >>> c = Counter(a=3, b=1)
263 >>> d = Counter(a=1, b=2)
Raymond Hettinger4d2073a2009-01-20 03:41:22 +0000264 >>> c + d # add two counters together: c[x] + d[x]
265 Counter({'a': 4, 'b': 3})
266 >>> c - d # subtract (keeping only positive counts)
267 Counter({'a': 2})
Raymond Hettinger7bf3a0e2009-01-20 07:15:22 +0000268 >>> c & d # intersection: min(c[x], d[x])
Raymond Hettinger4d2073a2009-01-20 03:41:22 +0000269 Counter({'a': 1, 'b': 1})
270 >>> c | d # union: max(c[x], d[x])
271 Counter({'a': 3, 'b': 2})
272
Raymond Hettingerb14043c2009-01-20 23:44:31 +0000273.. seealso::
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000274
Raymond Hettingerb14043c2009-01-20 23:44:31 +0000275 * `Bag class <http://www.gnu.org/software/smalltalk/manual-base/html_node/Bag.html>`_
276 in Smalltalk.
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000277
Raymond Hettinger670eaec2009-01-21 23:14:07 +0000278 * A `Counter <http://code.activestate.com/recipes/576611/>`_ conformant
279 recipe for Python 2.5 and an early Python `Bag recipe
280 <http://code.activestate.com/recipes/259174/>`_ for Python 2.4.
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000281
Raymond Hettingerb14043c2009-01-20 23:44:31 +0000282 * Wikipedia entry for `Multisets <http://en.wikipedia.org/wiki/Multiset>`_\.
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000283
Raymond Hettingerb14043c2009-01-20 23:44:31 +0000284 * `C++ multisets <http://www.demo2s.com/Tutorial/Cpp/0380__set-multiset/Catalog0380__set-multiset.htm>`_
285 tutorial with standalone examples.
286
287 * For use cases for multisets and mathematical operations on multisets, see
288 *Knuth, Donald. The Art of Computer Programming Volume II,
289 Section 4.6.3, Exercise 19*\.
290
Raymond Hettinger670eaec2009-01-21 23:14:07 +0000291 * To enumerate all distinct multisets of a given size over a given set of
292 elements, see the :func:`combinations_with_replacement` function in the
293 :ref:`itertools-recipes` for itertools::
Raymond Hettingerb14043c2009-01-20 23:44:31 +0000294
295 map(Counter, combinations_with_replacement('abc', 2)) --> AA AB AC BB BC CC
Raymond Hettingerb8baf632009-01-14 02:20:07 +0000296
297
Georg Brandl116aa622007-08-15 14:28:22 +0000298:class:`deque` objects
299----------------------
300
Georg Brandl9afde1c2007-11-01 20:32:30 +0000301.. class:: deque([iterable[, maxlen]])
Georg Brandl116aa622007-08-15 14:28:22 +0000302
303 Returns a new deque object initialized left-to-right (using :meth:`append`) with
304 data from *iterable*. If *iterable* is not specified, the new deque is empty.
305
306 Deques are a generalization of stacks and queues (the name is pronounced "deck"
307 and is short for "double-ended queue"). Deques support thread-safe, memory
308 efficient appends and pops from either side of the deque with approximately the
309 same O(1) performance in either direction.
310
311 Though :class:`list` objects support similar operations, they are optimized for
312 fast fixed-length operations and incur O(n) memory movement costs for
313 ``pop(0)`` and ``insert(0, v)`` operations which change both the size and
314 position of the underlying data representation.
315
Georg Brandl116aa622007-08-15 14:28:22 +0000316
Georg Brandl9afde1c2007-11-01 20:32:30 +0000317 If *maxlen* is not specified or is *None*, deques may grow to an
318 arbitrary length. Otherwise, the deque is bounded to the specified maximum
319 length. Once a bounded length deque is full, when new items are added, a
320 corresponding number of items are discarded from the opposite end. Bounded
321 length deques provide functionality similar to the ``tail`` filter in
322 Unix. They are also useful for tracking transactions and other pools of data
323 where only the most recent activity is of interest.
324
Georg Brandl9afde1c2007-11-01 20:32:30 +0000325
Benjamin Petersone41251e2008-04-25 01:59:09 +0000326 Deque objects support the following methods:
Georg Brandl116aa622007-08-15 14:28:22 +0000327
Benjamin Petersone41251e2008-04-25 01:59:09 +0000328 .. method:: append(x)
Georg Brandl116aa622007-08-15 14:28:22 +0000329
Benjamin Petersone41251e2008-04-25 01:59:09 +0000330 Add *x* to the right side of the deque.
Georg Brandl116aa622007-08-15 14:28:22 +0000331
332
Benjamin Petersone41251e2008-04-25 01:59:09 +0000333 .. method:: appendleft(x)
Georg Brandl116aa622007-08-15 14:28:22 +0000334
Benjamin Petersone41251e2008-04-25 01:59:09 +0000335 Add *x* to the left side of the deque.
Georg Brandl116aa622007-08-15 14:28:22 +0000336
337
Benjamin Petersone41251e2008-04-25 01:59:09 +0000338 .. method:: clear()
Georg Brandl116aa622007-08-15 14:28:22 +0000339
Benjamin Petersone41251e2008-04-25 01:59:09 +0000340 Remove all elements from the deque leaving it with length 0.
Georg Brandl116aa622007-08-15 14:28:22 +0000341
342
Benjamin Petersone41251e2008-04-25 01:59:09 +0000343 .. method:: extend(iterable)
Georg Brandl116aa622007-08-15 14:28:22 +0000344
Benjamin Petersone41251e2008-04-25 01:59:09 +0000345 Extend the right side of the deque by appending elements from the iterable
346 argument.
Georg Brandl116aa622007-08-15 14:28:22 +0000347
348
Benjamin Petersone41251e2008-04-25 01:59:09 +0000349 .. method:: extendleft(iterable)
Georg Brandl116aa622007-08-15 14:28:22 +0000350
Benjamin Petersone41251e2008-04-25 01:59:09 +0000351 Extend the left side of the deque by appending elements from *iterable*.
352 Note, the series of left appends results in reversing the order of
353 elements in the iterable argument.
Georg Brandl116aa622007-08-15 14:28:22 +0000354
355
Benjamin Petersone41251e2008-04-25 01:59:09 +0000356 .. method:: pop()
Georg Brandl116aa622007-08-15 14:28:22 +0000357
Benjamin Petersone41251e2008-04-25 01:59:09 +0000358 Remove and return an element from the right side of the deque. If no
359 elements are present, raises an :exc:`IndexError`.
Georg Brandl116aa622007-08-15 14:28:22 +0000360
361
Benjamin Petersone41251e2008-04-25 01:59:09 +0000362 .. method:: popleft()
Georg Brandl116aa622007-08-15 14:28:22 +0000363
Benjamin Petersone41251e2008-04-25 01:59:09 +0000364 Remove and return an element from the left side of the deque. If no
365 elements are present, raises an :exc:`IndexError`.
Georg Brandl116aa622007-08-15 14:28:22 +0000366
367
Benjamin Petersone41251e2008-04-25 01:59:09 +0000368 .. method:: remove(value)
Georg Brandl116aa622007-08-15 14:28:22 +0000369
Benjamin Petersone41251e2008-04-25 01:59:09 +0000370 Removed the first occurrence of *value*. If not found, raises a
371 :exc:`ValueError`.
Georg Brandl116aa622007-08-15 14:28:22 +0000372
Georg Brandl116aa622007-08-15 14:28:22 +0000373
Benjamin Petersone41251e2008-04-25 01:59:09 +0000374 .. method:: rotate(n)
Georg Brandl116aa622007-08-15 14:28:22 +0000375
Benjamin Petersone41251e2008-04-25 01:59:09 +0000376 Rotate the deque *n* steps to the right. If *n* is negative, rotate to
377 the left. Rotating one step to the right is equivalent to:
378 ``d.appendleft(d.pop())``.
379
Georg Brandl116aa622007-08-15 14:28:22 +0000380
381In addition to the above, deques support iteration, pickling, ``len(d)``,
382``reversed(d)``, ``copy.copy(d)``, ``copy.deepcopy(d)``, membership testing with
Benjamin Peterson206e3072008-10-19 14:07:49 +0000383the :keyword:`in` operator, and subscript references such as ``d[-1]``. Indexed
384access is O(1) at both ends but slows to O(n) in the middle. For fast random
385access, use lists instead.
Georg Brandl116aa622007-08-15 14:28:22 +0000386
Christian Heimesfe337bf2008-03-23 21:54:12 +0000387Example:
388
389.. doctest::
Georg Brandl116aa622007-08-15 14:28:22 +0000390
391 >>> from collections import deque
392 >>> d = deque('ghi') # make a new deque with three items
393 >>> for elem in d: # iterate over the deque's elements
Neal Norwitz752abd02008-05-13 04:55:24 +0000394 ... print(elem.upper())
Georg Brandl116aa622007-08-15 14:28:22 +0000395 G
396 H
397 I
398
399 >>> d.append('j') # add a new entry to the right side
400 >>> d.appendleft('f') # add a new entry to the left side
401 >>> d # show the representation of the deque
402 deque(['f', 'g', 'h', 'i', 'j'])
403
404 >>> d.pop() # return and remove the rightmost item
405 'j'
406 >>> d.popleft() # return and remove the leftmost item
407 'f'
408 >>> list(d) # list the contents of the deque
409 ['g', 'h', 'i']
410 >>> d[0] # peek at leftmost item
411 'g'
412 >>> d[-1] # peek at rightmost item
413 'i'
414
415 >>> list(reversed(d)) # list the contents of a deque in reverse
416 ['i', 'h', 'g']
417 >>> 'h' in d # search the deque
418 True
419 >>> d.extend('jkl') # add multiple elements at once
420 >>> d
421 deque(['g', 'h', 'i', 'j', 'k', 'l'])
422 >>> d.rotate(1) # right rotation
423 >>> d
424 deque(['l', 'g', 'h', 'i', 'j', 'k'])
425 >>> d.rotate(-1) # left rotation
426 >>> d
427 deque(['g', 'h', 'i', 'j', 'k', 'l'])
428
429 >>> deque(reversed(d)) # make a new deque in reverse order
430 deque(['l', 'k', 'j', 'i', 'h', 'g'])
431 >>> d.clear() # empty the deque
432 >>> d.pop() # cannot pop from an empty deque
433 Traceback (most recent call last):
434 File "<pyshell#6>", line 1, in -toplevel-
435 d.pop()
436 IndexError: pop from an empty deque
437
438 >>> d.extendleft('abc') # extendleft() reverses the input order
439 >>> d
440 deque(['c', 'b', 'a'])
441
442
Georg Brandl9afde1c2007-11-01 20:32:30 +0000443:class:`deque` Recipes
444^^^^^^^^^^^^^^^^^^^^^^
Georg Brandl116aa622007-08-15 14:28:22 +0000445
446This section shows various approaches to working with deques.
447
448The :meth:`rotate` method provides a way to implement :class:`deque` slicing and
449deletion. For example, a pure python implementation of ``del d[n]`` relies on
450the :meth:`rotate` method to position elements to be popped::
451
452 def delete_nth(d, n):
453 d.rotate(-n)
454 d.popleft()
455 d.rotate(n)
456
457To implement :class:`deque` slicing, use a similar approach applying
458:meth:`rotate` to bring a target element to the left side of the deque. Remove
459old entries with :meth:`popleft`, add new entries with :meth:`extend`, and then
460reverse the rotation.
Georg Brandl116aa622007-08-15 14:28:22 +0000461With minor variations on that approach, it is easy to implement Forth style
462stack manipulations such as ``dup``, ``drop``, ``swap``, ``over``, ``pick``,
463``rot``, and ``roll``.
464
Georg Brandl116aa622007-08-15 14:28:22 +0000465Multi-pass data reduction algorithms can be succinctly expressed and efficiently
466coded by extracting elements with multiple calls to :meth:`popleft`, applying
Georg Brandl9afde1c2007-11-01 20:32:30 +0000467a reduction function, and calling :meth:`append` to add the result back to the
468deque.
Georg Brandl116aa622007-08-15 14:28:22 +0000469
470For example, building a balanced binary tree of nested lists entails reducing
Christian Heimesfe337bf2008-03-23 21:54:12 +0000471two adjacent nodes into one by grouping them in a list:
Georg Brandl116aa622007-08-15 14:28:22 +0000472
473 >>> def maketree(iterable):
474 ... d = deque(iterable)
475 ... while len(d) > 1:
476 ... pair = [d.popleft(), d.popleft()]
477 ... d.append(pair)
478 ... return list(d)
479 ...
Georg Brandl6911e3c2007-09-04 07:15:32 +0000480 >>> print(maketree('abcdefgh'))
Georg Brandl116aa622007-08-15 14:28:22 +0000481 [[[['a', 'b'], ['c', 'd']], [['e', 'f'], ['g', 'h']]]]
482
Georg Brandl9afde1c2007-11-01 20:32:30 +0000483Bounded length deques provide functionality similar to the ``tail`` filter
484in Unix::
Georg Brandl116aa622007-08-15 14:28:22 +0000485
Georg Brandl9afde1c2007-11-01 20:32:30 +0000486 def tail(filename, n=10):
487 'Return the last n lines of a file'
488 return deque(open(filename), n)
Georg Brandl116aa622007-08-15 14:28:22 +0000489
Georg Brandl116aa622007-08-15 14:28:22 +0000490
491:class:`defaultdict` objects
492----------------------------
493
Georg Brandl116aa622007-08-15 14:28:22 +0000494.. class:: defaultdict([default_factory[, ...]])
495
496 Returns a new dictionary-like object. :class:`defaultdict` is a subclass of the
497 builtin :class:`dict` class. It overrides one method and adds one writable
498 instance variable. The remaining functionality is the same as for the
499 :class:`dict` class and is not documented here.
500
501 The first argument provides the initial value for the :attr:`default_factory`
502 attribute; it defaults to ``None``. All remaining arguments are treated the same
503 as if they were passed to the :class:`dict` constructor, including keyword
504 arguments.
505
Georg Brandl116aa622007-08-15 14:28:22 +0000506
Benjamin Petersone41251e2008-04-25 01:59:09 +0000507 :class:`defaultdict` objects support the following method in addition to the
508 standard :class:`dict` operations:
Georg Brandl116aa622007-08-15 14:28:22 +0000509
Benjamin Petersone41251e2008-04-25 01:59:09 +0000510 .. method:: defaultdict.__missing__(key)
Georg Brandl116aa622007-08-15 14:28:22 +0000511
Benjamin Peterson5478b472008-09-17 22:25:09 +0000512 If the :attr:`default_factory` attribute is ``None``, this raises a
Benjamin Petersone41251e2008-04-25 01:59:09 +0000513 :exc:`KeyError` exception with the *key* as argument.
Georg Brandl116aa622007-08-15 14:28:22 +0000514
Benjamin Petersone41251e2008-04-25 01:59:09 +0000515 If :attr:`default_factory` is not ``None``, it is called without arguments
516 to provide a default value for the given *key*, this value is inserted in
517 the dictionary for the *key*, and returned.
Georg Brandl116aa622007-08-15 14:28:22 +0000518
Benjamin Petersone41251e2008-04-25 01:59:09 +0000519 If calling :attr:`default_factory` raises an exception this exception is
520 propagated unchanged.
Georg Brandl116aa622007-08-15 14:28:22 +0000521
Benjamin Petersone41251e2008-04-25 01:59:09 +0000522 This method is called by the :meth:`__getitem__` method of the
523 :class:`dict` class when the requested key is not found; whatever it
524 returns or raises is then returned or raised by :meth:`__getitem__`.
Georg Brandl116aa622007-08-15 14:28:22 +0000525
526
Benjamin Petersone41251e2008-04-25 01:59:09 +0000527 :class:`defaultdict` objects support the following instance variable:
Georg Brandl116aa622007-08-15 14:28:22 +0000528
Benjamin Petersone41251e2008-04-25 01:59:09 +0000529
530 .. attribute:: defaultdict.default_factory
531
532 This attribute is used by the :meth:`__missing__` method; it is
533 initialized from the first argument to the constructor, if present, or to
534 ``None``, if absent.
Georg Brandl116aa622007-08-15 14:28:22 +0000535
536
Georg Brandl116aa622007-08-15 14:28:22 +0000537:class:`defaultdict` Examples
538^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
539
540Using :class:`list` as the :attr:`default_factory`, it is easy to group a
Christian Heimesfe337bf2008-03-23 21:54:12 +0000541sequence of key-value pairs into a dictionary of lists:
Georg Brandl116aa622007-08-15 14:28:22 +0000542
543 >>> s = [('yellow', 1), ('blue', 2), ('yellow', 3), ('blue', 4), ('red', 1)]
544 >>> d = defaultdict(list)
545 >>> for k, v in s:
546 ... d[k].append(v)
547 ...
548 >>> d.items()
549 [('blue', [2, 4]), ('red', [1]), ('yellow', [1, 3])]
550
551When each key is encountered for the first time, it is not already in the
552mapping; so an entry is automatically created using the :attr:`default_factory`
553function which returns an empty :class:`list`. The :meth:`list.append`
554operation then attaches the value to the new list. When keys are encountered
555again, the look-up proceeds normally (returning the list for that key) and the
556:meth:`list.append` operation adds another value to the list. This technique is
Christian Heimesfe337bf2008-03-23 21:54:12 +0000557simpler and faster than an equivalent technique using :meth:`dict.setdefault`:
Georg Brandl116aa622007-08-15 14:28:22 +0000558
559 >>> d = {}
560 >>> for k, v in s:
561 ... d.setdefault(k, []).append(v)
562 ...
563 >>> d.items()
564 [('blue', [2, 4]), ('red', [1]), ('yellow', [1, 3])]
565
566Setting the :attr:`default_factory` to :class:`int` makes the
567:class:`defaultdict` useful for counting (like a bag or multiset in other
Christian Heimesfe337bf2008-03-23 21:54:12 +0000568languages):
Georg Brandl116aa622007-08-15 14:28:22 +0000569
570 >>> s = 'mississippi'
571 >>> d = defaultdict(int)
572 >>> for k in s:
573 ... d[k] += 1
574 ...
575 >>> d.items()
576 [('i', 4), ('p', 2), ('s', 4), ('m', 1)]
577
578When a letter is first encountered, it is missing from the mapping, so the
579:attr:`default_factory` function calls :func:`int` to supply a default count of
580zero. The increment operation then builds up the count for each letter.
581
582The function :func:`int` which always returns zero is just a special case of
583constant functions. A faster and more flexible way to create constant functions
584is to use a lambda function which can supply any constant value (not just
Christian Heimesfe337bf2008-03-23 21:54:12 +0000585zero):
Georg Brandl116aa622007-08-15 14:28:22 +0000586
587 >>> def constant_factory(value):
588 ... return lambda: value
589 >>> d = defaultdict(constant_factory('<missing>'))
590 >>> d.update(name='John', action='ran')
591 >>> '%(name)s %(action)s to %(object)s' % d
592 'John ran to <missing>'
593
594Setting the :attr:`default_factory` to :class:`set` makes the
Christian Heimesfe337bf2008-03-23 21:54:12 +0000595:class:`defaultdict` useful for building a dictionary of sets:
Georg Brandl116aa622007-08-15 14:28:22 +0000596
597 >>> s = [('red', 1), ('blue', 2), ('red', 3), ('blue', 4), ('red', 1), ('blue', 4)]
598 >>> d = defaultdict(set)
599 >>> for k, v in s:
600 ... d[k].add(v)
601 ...
602 >>> d.items()
603 [('blue', set([2, 4])), ('red', set([1, 3]))]
604
605
Guido van Rossum3d392eb2007-11-16 00:35:22 +0000606:func:`namedtuple` Factory Function for Tuples with Named Fields
Christian Heimes790c8232008-01-07 21:14:23 +0000607----------------------------------------------------------------
Georg Brandl116aa622007-08-15 14:28:22 +0000608
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000609Named tuples assign meaning to each position in a tuple and allow for more readable,
610self-documenting code. They can be used wherever regular tuples are used, and
611they add the ability to access fields by name instead of position index.
Georg Brandl116aa622007-08-15 14:28:22 +0000612
Benjamin Peterson4469d0c2008-11-30 22:46:23 +0000613.. function:: namedtuple(typename, field_names, [verbose])
Georg Brandl116aa622007-08-15 14:28:22 +0000614
615 Returns a new tuple subclass named *typename*. The new subclass is used to
Christian Heimesc3f30c42008-02-22 16:37:40 +0000616 create tuple-like objects that have fields accessible by attribute lookup as
Georg Brandl116aa622007-08-15 14:28:22 +0000617 well as being indexable and iterable. Instances of the subclass also have a
Benjamin Peterson4469d0c2008-11-30 22:46:23 +0000618 helpful docstring (with typename and field_names) and a helpful :meth:`__repr__`
Georg Brandl116aa622007-08-15 14:28:22 +0000619 method which lists the tuple contents in a ``name=value`` format.
620
Benjamin Peterson4469d0c2008-11-30 22:46:23 +0000621 The *field_names* are a single string with each fieldname separated by whitespace
622 and/or commas, for example ``'x y'`` or ``'x, y'``. Alternatively, *field_names*
Christian Heimes25bb7832008-01-11 16:17:00 +0000623 can be a sequence of strings such as ``['x', 'y']``.
Georg Brandl9afde1c2007-11-01 20:32:30 +0000624
625 Any valid Python identifier may be used for a fieldname except for names
Christian Heimes0449f632007-12-15 01:27:15 +0000626 starting with an underscore. Valid identifiers consist of letters, digits,
627 and underscores but do not start with a digit or underscore and cannot be
Georg Brandlf6945182008-02-01 11:56:49 +0000628 a :mod:`keyword` such as *class*, *for*, *return*, *global*, *pass*,
Georg Brandl9afde1c2007-11-01 20:32:30 +0000629 or *raise*.
Georg Brandl116aa622007-08-15 14:28:22 +0000630
Christian Heimes25bb7832008-01-11 16:17:00 +0000631 If *verbose* is true, the class definition is printed just before being built.
Georg Brandl116aa622007-08-15 14:28:22 +0000632
Georg Brandl9afde1c2007-11-01 20:32:30 +0000633 Named tuple instances do not have per-instance dictionaries, so they are
Thomas Wouters8ce81f72007-09-20 18:22:40 +0000634 lightweight and require no more memory than regular tuples.
Georg Brandl116aa622007-08-15 14:28:22 +0000635
Christian Heimesfe337bf2008-03-23 21:54:12 +0000636Example:
637
638.. doctest::
639 :options: +NORMALIZE_WHITESPACE
Georg Brandl116aa622007-08-15 14:28:22 +0000640
Guido van Rossum3d392eb2007-11-16 00:35:22 +0000641 >>> Point = namedtuple('Point', 'x y', verbose=True)
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000642 class Point(tuple):
643 'Point(x, y)'
Christian Heimesfe337bf2008-03-23 21:54:12 +0000644 <BLANKLINE>
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000645 __slots__ = ()
Christian Heimesfe337bf2008-03-23 21:54:12 +0000646 <BLANKLINE>
Christian Heimesfaf2f632008-01-06 16:59:19 +0000647 _fields = ('x', 'y')
Christian Heimesfe337bf2008-03-23 21:54:12 +0000648 <BLANKLINE>
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000649 def __new__(cls, x, y):
650 return tuple.__new__(cls, (x, y))
Christian Heimesfe337bf2008-03-23 21:54:12 +0000651 <BLANKLINE>
Christian Heimesfaf2f632008-01-06 16:59:19 +0000652 @classmethod
Christian Heimesfe337bf2008-03-23 21:54:12 +0000653 def _make(cls, iterable, new=tuple.__new__, len=len):
Christian Heimesfaf2f632008-01-06 16:59:19 +0000654 'Make a new Point object from a sequence or iterable'
Christian Heimesfe337bf2008-03-23 21:54:12 +0000655 result = new(cls, iterable)
Christian Heimesfaf2f632008-01-06 16:59:19 +0000656 if len(result) != 2:
657 raise TypeError('Expected 2 arguments, got %d' % len(result))
658 return result
Christian Heimesfe337bf2008-03-23 21:54:12 +0000659 <BLANKLINE>
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000660 def __repr__(self):
661 return 'Point(x=%r, y=%r)' % self
Christian Heimesfe337bf2008-03-23 21:54:12 +0000662 <BLANKLINE>
Christian Heimes99170a52007-12-19 02:07:34 +0000663 def _asdict(t):
Christian Heimes0449f632007-12-15 01:27:15 +0000664 'Return a new dict which maps field names to their values'
Christian Heimes99170a52007-12-19 02:07:34 +0000665 return {'x': t[0], 'y': t[1]}
Christian Heimesfe337bf2008-03-23 21:54:12 +0000666 <BLANKLINE>
Christian Heimes0449f632007-12-15 01:27:15 +0000667 def _replace(self, **kwds):
Guido van Rossum3d392eb2007-11-16 00:35:22 +0000668 'Return a new Point object replacing specified fields with new values'
Christian Heimesfaf2f632008-01-06 16:59:19 +0000669 result = self._make(map(kwds.pop, ('x', 'y'), self))
670 if kwds:
671 raise ValueError('Got unexpected field names: %r' % kwds.keys())
672 return result
Georg Brandl48310cd2009-01-03 21:18:54 +0000673 <BLANKLINE>
674 def __getnewargs__(self):
Benjamin Peterson41181742008-07-02 20:22:54 +0000675 return tuple(self)
Christian Heimesfe337bf2008-03-23 21:54:12 +0000676 <BLANKLINE>
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000677 x = property(itemgetter(0))
678 y = property(itemgetter(1))
Georg Brandl116aa622007-08-15 14:28:22 +0000679
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000680 >>> p = Point(11, y=22) # instantiate with positional or keyword arguments
Christian Heimes99170a52007-12-19 02:07:34 +0000681 >>> p[0] + p[1] # indexable like the plain tuple (11, 22)
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000682 33
683 >>> x, y = p # unpack like a regular tuple
684 >>> x, y
685 (11, 22)
Christian Heimesc3f30c42008-02-22 16:37:40 +0000686 >>> p.x + p.y # fields also accessible by name
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000687 33
688 >>> p # readable __repr__ with a name=value style
689 Point(x=11, y=22)
Georg Brandl116aa622007-08-15 14:28:22 +0000690
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000691Named tuples are especially useful for assigning field names to result tuples returned
692by the :mod:`csv` or :mod:`sqlite3` modules::
693
Guido van Rossum3d392eb2007-11-16 00:35:22 +0000694 EmployeeRecord = namedtuple('EmployeeRecord', 'name, age, title, department, paygrade')
Georg Brandl9afde1c2007-11-01 20:32:30 +0000695
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000696 import csv
Christian Heimesfaf2f632008-01-06 16:59:19 +0000697 for emp in map(EmployeeRecord._make, csv.reader(open("employees.csv", "rb"))):
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000698 print(emp.name, emp.title)
699
Georg Brandl9afde1c2007-11-01 20:32:30 +0000700 import sqlite3
701 conn = sqlite3.connect('/companydata')
702 cursor = conn.cursor()
703 cursor.execute('SELECT name, age, title, department, paygrade FROM employees')
Christian Heimesfaf2f632008-01-06 16:59:19 +0000704 for emp in map(EmployeeRecord._make, cursor.fetchall()):
Christian Heimes00412232008-01-10 16:02:19 +0000705 print(emp.name, emp.title)
Georg Brandl9afde1c2007-11-01 20:32:30 +0000706
Christian Heimes99170a52007-12-19 02:07:34 +0000707In addition to the methods inherited from tuples, named tuples support
Christian Heimes2380ac72008-01-09 00:17:24 +0000708three additional methods and one attribute. To prevent conflicts with
709field names, the method and attribute names start with an underscore.
Christian Heimes99170a52007-12-19 02:07:34 +0000710
Christian Heimes790c8232008-01-07 21:14:23 +0000711.. method:: somenamedtuple._make(iterable)
Christian Heimes99170a52007-12-19 02:07:34 +0000712
Christian Heimesfaf2f632008-01-06 16:59:19 +0000713 Class method that makes a new instance from an existing sequence or iterable.
Christian Heimes99170a52007-12-19 02:07:34 +0000714
Christian Heimesfe337bf2008-03-23 21:54:12 +0000715.. doctest::
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000716
Christian Heimesfaf2f632008-01-06 16:59:19 +0000717 >>> t = [11, 22]
718 >>> Point._make(t)
719 Point(x=11, y=22)
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000720
Christian Heimes790c8232008-01-07 21:14:23 +0000721.. method:: somenamedtuple._asdict()
Georg Brandl9afde1c2007-11-01 20:32:30 +0000722
Christian Heimesfe337bf2008-03-23 21:54:12 +0000723 Return a new dict which maps field names to their corresponding values::
Georg Brandl9afde1c2007-11-01 20:32:30 +0000724
Christian Heimes0449f632007-12-15 01:27:15 +0000725 >>> p._asdict()
Georg Brandl9afde1c2007-11-01 20:32:30 +0000726 {'x': 11, 'y': 22}
Christian Heimesfe337bf2008-03-23 21:54:12 +0000727
Christian Heimes790c8232008-01-07 21:14:23 +0000728.. method:: somenamedtuple._replace(kwargs)
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000729
Christian Heimesfe337bf2008-03-23 21:54:12 +0000730 Return a new instance of the named tuple replacing specified fields with new
731 values:
Thomas Wouters8ce81f72007-09-20 18:22:40 +0000732
733::
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000734
735 >>> p = Point(x=11, y=22)
Christian Heimes0449f632007-12-15 01:27:15 +0000736 >>> p._replace(x=33)
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000737 Point(x=33, y=22)
738
Guido van Rossum3d392eb2007-11-16 00:35:22 +0000739 >>> for partnum, record in inventory.items():
Christian Heimes454f37b2008-01-10 00:10:02 +0000740 ... inventory[partnum] = record._replace(price=newprices[partnum], timestamp=time.now())
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000741
Christian Heimes790c8232008-01-07 21:14:23 +0000742.. attribute:: somenamedtuple._fields
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000743
Christian Heimes2380ac72008-01-09 00:17:24 +0000744 Tuple of strings listing the field names. Useful for introspection
Georg Brandl9afde1c2007-11-01 20:32:30 +0000745 and for creating new named tuple types from existing named tuples.
Thomas Wouters8ce81f72007-09-20 18:22:40 +0000746
Christian Heimesfe337bf2008-03-23 21:54:12 +0000747.. doctest::
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000748
Christian Heimes0449f632007-12-15 01:27:15 +0000749 >>> p._fields # view the field names
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000750 ('x', 'y')
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000751
Guido van Rossum3d392eb2007-11-16 00:35:22 +0000752 >>> Color = namedtuple('Color', 'red green blue')
Christian Heimes0449f632007-12-15 01:27:15 +0000753 >>> Pixel = namedtuple('Pixel', Point._fields + Color._fields)
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000754 >>> Pixel(11, 22, 128, 255, 0)
Christian Heimes454f37b2008-01-10 00:10:02 +0000755 Pixel(x=11, y=22, red=128, green=255, blue=0)
Georg Brandl116aa622007-08-15 14:28:22 +0000756
Christian Heimes0449f632007-12-15 01:27:15 +0000757To retrieve a field whose name is stored in a string, use the :func:`getattr`
Christian Heimesfe337bf2008-03-23 21:54:12 +0000758function:
Christian Heimes0449f632007-12-15 01:27:15 +0000759
760 >>> getattr(p, 'x')
761 11
762
Christian Heimesfe337bf2008-03-23 21:54:12 +0000763To convert a dictionary to a named tuple, use the double-star-operator [#]_:
Christian Heimes99170a52007-12-19 02:07:34 +0000764
765 >>> d = {'x': 11, 'y': 22}
766 >>> Point(**d)
767 Point(x=11, y=22)
768
Guido van Rossum3d392eb2007-11-16 00:35:22 +0000769Since a named tuple is a regular Python class, it is easy to add or change
Christian Heimes043d6f62008-01-07 17:19:16 +0000770functionality with a subclass. Here is how to add a calculated field and
Christian Heimesfe337bf2008-03-23 21:54:12 +0000771a fixed-width print format:
Guido van Rossum3d392eb2007-11-16 00:35:22 +0000772
Christian Heimes043d6f62008-01-07 17:19:16 +0000773 >>> class Point(namedtuple('Point', 'x y')):
Christian Heimes25bb7832008-01-11 16:17:00 +0000774 ... __slots__ = ()
Christian Heimes454f37b2008-01-10 00:10:02 +0000775 ... @property
776 ... def hypot(self):
777 ... return (self.x ** 2 + self.y ** 2) ** 0.5
778 ... def __str__(self):
Christian Heimes25bb7832008-01-11 16:17:00 +0000779 ... 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 +0000780
Georg Brandl0df79792008-10-04 18:33:26 +0000781 >>> for p in Point(3, 4), Point(14, 5/7):
Christian Heimes00412232008-01-10 16:02:19 +0000782 ... print(p)
Christian Heimes25bb7832008-01-11 16:17:00 +0000783 Point: x= 3.000 y= 4.000 hypot= 5.000
784 Point: x=14.000 y= 0.714 hypot=14.018
Christian Heimes043d6f62008-01-07 17:19:16 +0000785
Christian Heimesaf98da12008-01-27 15:18:18 +0000786The subclass shown above sets ``__slots__`` to an empty tuple. This keeps
Christian Heimes679db4a2008-01-18 09:56:22 +0000787keep memory requirements low by preventing the creation of instance dictionaries.
788
Christian Heimes2380ac72008-01-09 00:17:24 +0000789
790Subclassing is not useful for adding new, stored fields. Instead, simply
Christian Heimesfe337bf2008-03-23 21:54:12 +0000791create a new named tuple type from the :attr:`_fields` attribute:
Christian Heimes2380ac72008-01-09 00:17:24 +0000792
Christian Heimes25bb7832008-01-11 16:17:00 +0000793 >>> Point3D = namedtuple('Point3D', Point._fields + ('z',))
Christian Heimes2380ac72008-01-09 00:17:24 +0000794
795Default values can be implemented by using :meth:`_replace` to
Christian Heimesfe337bf2008-03-23 21:54:12 +0000796customize a prototype instance:
Guido van Rossum3d392eb2007-11-16 00:35:22 +0000797
798 >>> Account = namedtuple('Account', 'owner balance transaction_count')
Christian Heimes587c2bf2008-01-19 16:21:02 +0000799 >>> default_account = Account('<owner name>', 0.0, 0)
800 >>> johns_account = default_account._replace(owner='John')
Guido van Rossum3d392eb2007-11-16 00:35:22 +0000801
Christian Heimese4ca8152008-05-08 17:18:53 +0000802Enumerated constants can be implemented with named tuples, but it is simpler
803and more efficient to use a simple class declaration:
804
805 >>> Status = namedtuple('Status', 'open pending closed')._make(range(3))
806 >>> Status.open, Status.pending, Status.closed
807 (0, 1, 2)
808 >>> class Status:
809 ... open, pending, closed = range(3)
810
Thomas Wouters47b49bf2007-08-30 22:15:33 +0000811.. rubric:: Footnotes
812
Christian Heimes99170a52007-12-19 02:07:34 +0000813.. [#] For information on the double-star-operator see
Thomas Wouters47b49bf2007-08-30 22:15:33 +0000814 :ref:`tut-unpacking-arguments` and :ref:`calls`.
Raymond Hettingere4c96ad2008-02-06 01:23:58 +0000815
816
817
818:class:`UserDict` objects
Mark Summerfield8f2d0062008-02-06 13:30:44 +0000819-------------------------
Raymond Hettingere4c96ad2008-02-06 01:23:58 +0000820
Georg Brandl48310cd2009-01-03 21:18:54 +0000821The class, :class:`UserDict` acts as a wrapper around dictionary objects.
822The need for this class has been partially supplanted by the ability to
Raymond Hettingere4c96ad2008-02-06 01:23:58 +0000823subclass directly from :class:`dict`; however, this class can be easier
824to work with because the underlying dictionary is accessible as an
825attribute.
826
827.. class:: UserDict([initialdata])
828
829 Class that simulates a dictionary. The instance's contents are kept in a
830 regular dictionary, which is accessible via the :attr:`data` attribute of
831 :class:`UserDict` instances. If *initialdata* is provided, :attr:`data` is
832 initialized with its contents; note that a reference to *initialdata* will not
833 be kept, allowing it be used for other purposes.
834
Georg Brandl48310cd2009-01-03 21:18:54 +0000835In addition to supporting the methods and operations of mappings,
Raymond Hettingerebcee3f2008-02-06 19:54:00 +0000836:class:`UserDict` instances provide the following attribute:
Raymond Hettingere4c96ad2008-02-06 01:23:58 +0000837
838.. attribute:: UserDict.data
839
840 A real dictionary used to store the contents of the :class:`UserDict` class.
Raymond Hettinger53dbe392008-02-12 20:03:09 +0000841
842
843
844:class:`UserList` objects
845-------------------------
846
847This class acts as a wrapper around list objects. It is a useful base class
Georg Brandl48310cd2009-01-03 21:18:54 +0000848for your own list-like classes which can inherit from them and override
Raymond Hettinger53dbe392008-02-12 20:03:09 +0000849existing methods or add new ones. In this way, one can add new behaviors to
850lists.
851
Georg Brandl48310cd2009-01-03 21:18:54 +0000852The need for this class has been partially supplanted by the ability to
Raymond Hettinger53dbe392008-02-12 20:03:09 +0000853subclass directly from :class:`list`; however, this class can be easier
854to work with because the underlying list is accessible as an attribute.
855
856.. class:: UserList([list])
857
858 Class that simulates a list. The instance's contents are kept in a regular
859 list, which is accessible via the :attr:`data` attribute of :class:`UserList`
860 instances. The instance's contents are initially set to a copy of *list*,
861 defaulting to the empty list ``[]``. *list* can be any iterable, for
862 example a real Python list or a :class:`UserList` object.
863
Georg Brandl48310cd2009-01-03 21:18:54 +0000864In addition to supporting the methods and operations of mutable sequences,
Raymond Hettinger53dbe392008-02-12 20:03:09 +0000865:class:`UserList` instances provide the following attribute:
866
867.. attribute:: UserList.data
868
Georg Brandl48310cd2009-01-03 21:18:54 +0000869 A real :class:`list` object used to store the contents of the
Raymond Hettinger53dbe392008-02-12 20:03:09 +0000870 :class:`UserList` class.
871
872**Subclassing requirements:** Subclasses of :class:`UserList` are expect to
873offer a constructor which can be called with either no arguments or one
874argument. List operations which return a new sequence attempt to create an
875instance of the actual implementation class. To do so, it assumes that the
876constructor can be called with a single parameter, which is a sequence object
877used as a data source.
878
879If a derived class does not wish to comply with this requirement, all of the
880special methods supported by this class will need to be overridden; please
881consult the sources for information about the methods which need to be provided
882in that case.
Raymond Hettingerb3a65f82008-02-21 22:11:37 +0000883
884:class:`UserString` objects
Christian Heimesc3f30c42008-02-22 16:37:40 +0000885---------------------------
Raymond Hettingerb3a65f82008-02-21 22:11:37 +0000886
Georg Brandl48310cd2009-01-03 21:18:54 +0000887The class, :class:`UserString` acts as a wrapper around string objects.
888The need for this class has been partially supplanted by the ability to
Raymond Hettingerb3a65f82008-02-21 22:11:37 +0000889subclass directly from :class:`str`; however, this class can be easier
890to work with because the underlying string is accessible as an
891attribute.
892
893.. class:: UserString([sequence])
894
895 Class that simulates a string or a Unicode string object. The instance's
Georg Brandl48310cd2009-01-03 21:18:54 +0000896 content is kept in a regular string object, which is accessible via the
897 :attr:`data` attribute of :class:`UserString` instances. The instance's
Raymond Hettingerb3a65f82008-02-21 22:11:37 +0000898 contents are initially set to a copy of *sequence*. The *sequence* can
899 be an instance of :class:`bytes`, :class:`str`, :class:`UserString` (or a
900 subclass) or an arbitrary sequence which can be converted into a string using
901 the built-in :func:`str` function.