Python bisect模块的使用

Python bisect模块的使用

一、bisect模块简单介绍：

Python的bisect模块是内置模块，用于维护有序列表，它采用二分法来排序插入

bisect有哪些方法：

# 首先导入bisect包
import bisect
print(dir(bisect))
# 输出结果：
['__builtins__', '__cached__', '__doc__', '__file__',
 '__loader__', '__name__', '__package__', '__spec__',
 'bisect', 'bisect_left', 'bisect_right', 'insort',
 'insort_left', 'insort_right']

上面这些方法中，除了魔法方法，对另一些进行详细的讲解：

bisect：返回插入元素在列表中的下标

my_list = [1, 3, 5, 7, 9]
result = bisect.bisect(my_list, 6)
print(result)

# 输出结果：3 表示插入元素在列表中的下标

bisect_left：默认将元素查到左边，返回插入到左边的下标

my_list = [1, 3, 5, 7, 9]
result = bisect.bisect_left(my_list, 6)
print(result)

# 输出结果： 3 同bisect方法

bisect_right：与bisect_left相反

my_list = [1, 5, 3, 9, 7]
result = bisect.bisect_right(my_list, 6)
print(result)

# 输出结果： 3 

insort：会在列表中插入元素到正确位置

# 若列表中的元素是有序的话
my_list = [1, 3, 5, 7, 9]
bisect.insort(my_list, 2)
print(my_list)

# 输出结果：[1, 2, 3, 5, 7, 9]

# 若列表中的元素是无序的话，默认插入到最右边
my_list = [2, 3, 1, 5, 7, 4]
bisect.insort(my_list, 6)
print(my_list)

# 输出结果：[2, 3, 1, 5, 7, 4, 6]

insort_left：左边插入值，返回值None

my_list = [1, 3, 5, 7, 9]
bisect.insort_left(my_list, 2)
print(my_list)

# 输出结果：[1, 2, 3, 5, 7, 9]

insort_right：右边插入值，返回None

my_list = [1, 3, 5, 7, 9]
bisect.insort_right(my_list, 6)
print(my_list)

# 输出结果：[1, 3, 5, 6, 7, 9]

insort_left与insort_right：只是插入的位置不同而已

二、使用场景：

举个栗子，要根据学生的成绩进行分级，学生成绩(0-100)，分级成(ABCDEF)：

常规操作：

def grade(score):
    if score < 60:
        return 'F'
    elif score < 70:
        return 'E'
    elif score < 80:
        return 'D'
    elif score < 90:
        return 'C'
    elif score < 100:
        return 'B'
    else:
        return 'A'


scores = [59, 60, 72, 82, 85, 90, 98, 100]
result = [grade(item) for item in scores]
print(result)

# 输出结果如下:
['F', 'E', 'D', 'C', 'C', 'B', 'B', 'A']

使用bisect的操作：

• 比如学生成绩分级的分界点是[60, 70, 80, 90, 100]，等级是'FEDCBA'

  def grade(score, points=[60, 70, 80, 90, 100], grade="FEDCBA"):
      item = bisect.bisect(points, score)
      return grade[item]
  
  
  scores = [101, 60, 72, 82, 85, 90, 98, 100]
  result = [grade(item) for item in scores]
  print(result)
  
  # 输出结果如下：
  ['A', 'E', 'D', 'C', 'C', 'B', 'B', 'A']
  

• 商场会员积分来进行分级：

  def grade(score, points=[60, 70, 80, 90, 100], grade="FEDCBA"):
      item = bisect.bisect(points, score)
      print(grade[item])
      return grade[item]
  
  
  scores = [1000, 2000, 3000]
  names = ['普通会员', '中级会员', '高级会员', 'VIP']
  values = [999, 1999, 2999, 3001]
  print([grade(item, points=scores, grade=names) for item in values])
  
  # 输出结果如下：['普通会员', '中级会员', '高级会员', 'VIP']