Fred Drake | 295da24 | 1998-08-10 19:42:37 +0000 | [diff] [blame] | 1 | \section{\module{bisect} --- |
Fred Drake | bbac432 | 1999-02-20 00:14:17 +0000 | [diff] [blame] | 2 | Array bisection algorithm} |
| 3 | |
Fred Drake | b91e934 | 1998-07-23 17:59:49 +0000 | [diff] [blame] | 4 | \declaremodule{standard}{bisect} |
Fred Drake | edf6b1f | 1998-07-27 22:16:46 +0000 | [diff] [blame] | 5 | \modulesynopsis{Array bisection algorithms for binary searching.} |
Fred Drake | 38e5d27 | 2000-04-03 20:13:55 +0000 | [diff] [blame] | 6 | \sectionauthor{Fred L. Drake, Jr.}{fdrake@acm.org} |
| 7 | % LaTeX produced by Fred L. Drake, Jr. <fdrake@acm.org>, with an |
| 8 | % example based on the PyModules FAQ entry by Aaron Watters |
| 9 | % <arw@pythonpros.com>. |
Fred Drake | b91e934 | 1998-07-23 17:59:49 +0000 | [diff] [blame] | 10 | |
Fred Drake | ca6b4fe | 1998-04-28 18:28:21 +0000 | [diff] [blame] | 11 | |
Fred Drake | ca6b4fe | 1998-04-28 18:28:21 +0000 | [diff] [blame] | 12 | This module provides support for maintaining a list in sorted order |
| 13 | without having to sort the list after each insertion. For long lists |
| 14 | of items with expensive comparison operations, this can be an |
| 15 | improvement over the more common approach. The module is called |
| 16 | \module{bisect} because it uses a basic bisection algorithm to do its |
Fred Drake | d8a41e6 | 1999-02-19 17:54:10 +0000 | [diff] [blame] | 17 | work. The source code may be most useful as a working example of the |
Fred Drake | 91f2f26 | 2001-07-06 19:28:48 +0000 | [diff] [blame] | 18 | algorithm (the boundary conditions are already right!). |
Fred Drake | ca6b4fe | 1998-04-28 18:28:21 +0000 | [diff] [blame] | 19 | |
| 20 | The following functions are provided: |
| 21 | |
Fred Drake | 2a72712 | 2001-01-04 05:12:52 +0000 | [diff] [blame] | 22 | \begin{funcdesc}{bisect_left}{list, item\optional{, lo\optional{, hi}}} |
| 23 | Locate the proper insertion point for \var{item} in \var{list} to |
| 24 | maintain sorted order. The parameters \var{lo} and \var{hi} may be |
| 25 | used to specify a subset of the list which should be considered; by |
| 26 | default the entire list is used. If \var{item} is already present |
| 27 | in \var{list}, the insertion point will be before (to the left of) |
| 28 | any existing entries. The return value is suitable for use as the |
| 29 | first parameter to \code{\var{list}.insert()}. This assumes that |
| 30 | \var{list} is already sorted. |
| 31 | \versionadded{2.1} |
Fred Drake | ca6b4fe | 1998-04-28 18:28:21 +0000 | [diff] [blame] | 32 | \end{funcdesc} |
| 33 | |
Fred Drake | 2a72712 | 2001-01-04 05:12:52 +0000 | [diff] [blame] | 34 | \begin{funcdesc}{bisect_right}{list, item\optional{, lo\optional{, hi}}} |
| 35 | Similar to \function{bisect_left()}, but returns an insertion point |
| 36 | which comes after (to the right of) any existing entries of |
| 37 | \var{item} in \var{list}. |
| 38 | \versionadded{2.1} |
| 39 | \end{funcdesc} |
| 40 | |
| 41 | \begin{funcdesc}{bisect}{\unspecified} |
Fred Drake | 4e18f07 | 2001-01-04 14:18:55 +0000 | [diff] [blame] | 42 | Alias for \function{bisect_right()}. |
Fred Drake | 2a72712 | 2001-01-04 05:12:52 +0000 | [diff] [blame] | 43 | \end{funcdesc} |
| 44 | |
| 45 | \begin{funcdesc}{insort_left}{list, item\optional{, lo\optional{, hi}}} |
| 46 | Insert \var{item} in \var{list} in sorted order. This is equivalent |
| 47 | to \code{\var{list}.insert(bisect.bisect_left(\var{list}, \var{item}, |
| 48 | \var{lo}, \var{hi}), \var{item})}. This assumes that \var{list} is |
| 49 | already sorted. |
| 50 | \versionadded{2.1} |
| 51 | \end{funcdesc} |
| 52 | |
| 53 | \begin{funcdesc}{insort_right}{list, item\optional{, lo\optional{, hi}}} |
| 54 | Similar to \function{insort_left()}, but inserting \var{item} in |
| 55 | \var{list} after any existing entries of \var{item}. |
| 56 | \versionadded{2.1} |
| 57 | \end{funcdesc} |
| 58 | |
| 59 | \begin{funcdesc}{insort}{\unspecified} |
Fred Drake | 4e18f07 | 2001-01-04 14:18:55 +0000 | [diff] [blame] | 60 | Alias for \function{insort_right()}. |
Fred Drake | ca6b4fe | 1998-04-28 18:28:21 +0000 | [diff] [blame] | 61 | \end{funcdesc} |
| 62 | |
| 63 | |
Skip Montanaro | 09d9f86 | 2002-06-26 05:07:28 +0000 | [diff] [blame] | 64 | \subsection{Examples} |
Fred Drake | ca6b4fe | 1998-04-28 18:28:21 +0000 | [diff] [blame] | 65 | \nodename{bisect-example} |
| 66 | |
| 67 | The \function{bisect()} function is generally useful for categorizing |
Fred Drake | 4e18f07 | 2001-01-04 14:18:55 +0000 | [diff] [blame] | 68 | numeric data. This example uses \function{bisect()} to look up a |
Fred Drake | ca6b4fe | 1998-04-28 18:28:21 +0000 | [diff] [blame] | 69 | letter grade for an exam total (say) based on a set of ordered numeric |
| 70 | breakpoints: 85 and up is an `A', 75..84 is a `B', etc. |
| 71 | |
| 72 | \begin{verbatim} |
| 73 | >>> grades = "FEDCBA" |
| 74 | >>> breakpoints = [30, 44, 66, 75, 85] |
Fred Drake | 4e18f07 | 2001-01-04 14:18:55 +0000 | [diff] [blame] | 75 | >>> from bisect import bisect |
Fred Drake | ca6b4fe | 1998-04-28 18:28:21 +0000 | [diff] [blame] | 76 | >>> def grade(total): |
Fred Drake | 4e18f07 | 2001-01-04 14:18:55 +0000 | [diff] [blame] | 77 | ... return grades[bisect(breakpoints, total)] |
Fred Drake | ca6b4fe | 1998-04-28 18:28:21 +0000 | [diff] [blame] | 78 | ... |
| 79 | >>> grade(66) |
| 80 | 'C' |
| 81 | >>> map(grade, [33, 99, 77, 44, 12, 88]) |
| 82 | ['E', 'A', 'B', 'D', 'F', 'A'] |
| 83 | \end{verbatim} |
Skip Montanaro | 09d9f86 | 2002-06-26 05:07:28 +0000 | [diff] [blame] | 84 | |
| 85 | The bisect module can be used with the Queue module to implement a priority |
| 86 | queue (example courtesy of Fredrik Lundh): \index{Priority Queue} |
| 87 | |
| 88 | \begin{verbatim} |
| 89 | import Queue, bisect |
| 90 | |
| 91 | class PriorityQueue(Queue.Queue): |
| 92 | def _put(self, item): |
| 93 | bisect.insort(self.queue, item) |
| 94 | |
| 95 | # usage |
| 96 | queue = PriorityQueue(0) |
| 97 | queue.put((2, "second")) |
| 98 | queue.put((1, "first")) |
| 99 | queue.put((3, "third")) |
| 100 | priority, value = queue.get() |
| 101 | \end{verbatim} |