有没有更好的方法来检查一个数字是否是两个数字的范围

2022-02-24 00:00:00 python python-3.x if-statement range

问题描述

我正在尝试检查一个数字是否在整数范围内,并根据它所在的范围返回一个数字。我想知道有没有更好、更有效的方法来做这件事:

def checkRange(number):
    if number in range(0, 5499):
        return 5000
    elif number in range(5500, 9499):
        return 10000
    elif number in range(9500, 14499):
        return 15000
    elif number in range(14500, 19499):
        return 20000
    elif number in range(19500, 24499):
        return 25000
    elif number in range(24500, 29499):
        return 30000
    elif number in range(29500, 34499):
        return 35000
    elif number in range(34500, 39499):
        return 40000
    elif number in range(39500, 44499):
        return 45000

这感觉像是浪费资源,如果有更好的方法,我将不胜感激。


解决方案

由于您有连续的排序区域,因此执行此操作的一种更快且不太冗长的方法是使用bisect模块在断点列表中查找索引,然后使用它从值列表中获取相应值:

import bisect

break_points = [5499,  9499, 14499, 19499, 24499, 29499, 34499, 39499, 44499]
values       = [5000, 10000, 15000, 20000, 25000, 30000, 35000, 40000, 45000]

n = 10000
index = bisect.bisect_left(break_points, n)

values[index]
# 15000

如果有可能,您需要测试超过上一个断点的n值。或者,您可以将默认值添加到值列表的末尾。

相关文章