GeezerVison is showing github.com — fluentpythoncode/02-array-seq/bisect_demo.py at master · hellowac/fluentpythoncode

Expand file tree

bisect_demo.py

Copy path

More file actions

27 lines (20 loc) · 721 Bytes

Copy path

27 lines (20 loc) · 721 Bytes

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

# BEGIN BISECT_DEMO

import bisect

import sys

HAYSTACK = [1, 4, 5, 6, 8, 12, 15, 20, 21, 23, 23, 26, 29, 30]

NEEDLES = [0, 1, 2, 5, 8, 10, 22, 23, 29, 30, 31]

ROW_FMT = '{0:2d} @ {1:2d} {2}{0:<2d}'

def demo(bisect_fn):

for needle in reversed(NEEDLES):

position = bisect_fn(HAYSTACK, needle) # <1>

offset = position * ' |' # <2>

print(ROW_FMT.format(needle, position, offset)) # <3>

if __name__ == '__main__':

if sys.argv[-1] == 'left': # <4>

bisect_fn = bisect.bisect_left

else:

bisect_fn = bisect.bisect

print('DEMO:', bisect_fn.__name__) # <5>

print('haystack ->', ' '.join('%2d' % n for n in HAYSTACK))

demo(bisect_fn)

# END BISECT_DEMO