FilesExpand file tree

 master

/

bisect_demo.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

65 lines (50 loc) · 1.7 KB

 master

/

bisect_demo.py

Copy path

Top

File metadata and controls

• Code
• Blame

65 lines (50 loc) · 1.7 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

"""

bisect_demo.py

Demonstration of ``bisect.bisect``::

>>> import bisect

>>> demo(bisect.bisect)

31 @ 14 | | | | | | | | | | | | | |31

30 @ 14 | | | | | | | | | | | | | |30

29 @ 13 | | | | | | | | | | | | |29

23 @ 11 | | | | | | | | | | |23

22 @ 9 | | | | | | | | |22

10 @ 5 | | | | |10

8 @ 5 | | | | |8

5 @ 3 | | |5

2 @ 1 |2

1 @ 1 |1

0 @ 0 0

Demonstration of ``bisect.bisect_left``::

>>> demo(bisect.bisect_left)

31 @ 14 | | | | | | | | | | | | | |31

30 @ 13 | | | | | | | | | | | | |30

29 @ 12 | | | | | | | | | | | |29

23 @ 9 | | | | | | | | |23

22 @ 9 | | | | | | | | |22

10 @ 5 | | | | |10

8 @ 4 | | | |8

5 @ 2 | |5

2 @ 1 |2

1 @ 0 1

0 @ 0 0

"""

# tag::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(f'{n:2}' for n in HAYSTACK))

demo(bisect_fn)

# end::BISECT_DEMO[]