diff options
author | Raymond Hettinger <python@rcn.com> | 2010-08-07 07:36:55 (GMT) |
---|---|---|
committer | Raymond Hettinger <python@rcn.com> | 2010-08-07 07:36:55 (GMT) |
commit | 87c9d6cf9cbf2cada6a309cf4e46d77c3484b4c2 (patch) | |
tree | f4b7f9ef294b8ef93e1e2550e30688ab45ab7917 /Doc | |
parent | 1b669962815db3e0acc9a9f1f2a923be718350e5 (diff) | |
download | cpython-87c9d6cf9cbf2cada6a309cf4e46d77c3484b4c2.zip cpython-87c9d6cf9cbf2cada6a309cf4e46d77c3484b4c2.tar.gz cpython-87c9d6cf9cbf2cada6a309cf4e46d77c3484b4c2.tar.bz2 |
Improve the docs for bisect to cover common searching tasks.
Diffstat (limited to 'Doc')
-rw-r--r-- | Doc/library/bisect.rst | 60 |
1 files changed, 57 insertions, 3 deletions
diff --git a/Doc/library/bisect.rst b/Doc/library/bisect.rst index 3890d68..bcbcb07 100644 --- a/Doc/library/bisect.rst +++ b/Doc/library/bisect.rst @@ -39,6 +39,9 @@ The following functions are provided: ``a.insert(bisect.bisect_left(a, x, lo, hi), x)``. This assumes that *a* is already sorted. + Also note that while the fast search step is O(log n), the slower insertion + step is O(n), so the overall operation is slow. + .. function:: insort_right(a, x, lo=0, hi=len(a)) insort(a, x, lo=0, hi=len(a)) @@ -46,9 +49,53 @@ The following functions are provided: Similar to :func:`insort_left`, but inserting *x* in *a* after any existing entries of *x*. - -Examples --------- + Also note that while the fast search step is O(log n), the slower insertion + step is O(n), so the overall operation is slow. + +Searching Sorted Lists +---------------------- + +The above :func:`bisect` functions are useful for finding insertion points, but +can be tricky or awkward to use for common searching tasks. The following three +functions show how to transform them into the standard lookups for sorted +lists:: + + def find(a, key): + '''Find item with a key-value equal to key. + Raise ValueError if no such item exists. + + ''' + i = bisect_left(a, key) + if i < len(a) and a[i] == key: + return a[i] + raise ValueError('No item found with key equal to: %r' % (key,)) + + def find_le(a, key): + '''Find largest item with a key-value less-than or equal to key. + Raise ValueError if no such item exists. + If multiple key-values are equal, return the leftmost. + + ''' + i = bisect_left(a, key) + if i < len(a) and a[i] == key: + return a[i] + if i == 0: + raise ValueError('No item found with key at or below: %r' % (key,)) + return a[i-1] + + def find_ge(a, key): + '''Find smallest item with a key-value greater-than or equal to key. + Raise ValueError if no such item exists. + If multiple key-values are equal, return the leftmost. + + ''' + i = bisect_left(a, key) + if i == len(a): + raise ValueError('No item found with key at or above: %r' % (key,)) + return a[i] + +Other Examples +-------------- .. _bisect-example: @@ -87,3 +134,10 @@ of the record in question:: ('red', 5) >>> data[bisect_left(keys, 8)] ('yellow', 8) + +.. seealso:: + + `SortedCollection recipe + <http://code.activestate.com/recipes/577197-sortedcollection/>`_ that + encapsulates precomputed keys, allowing straight-forward insertion and + searching using a *key* function. |