Python3.x 有没有更好的方法来判断一个数字是否是两个数字的范围

发布于05月31日

我试图判断一个数字是否在整数范围内，并根据它所在的范围返回一个数字.我想知道是否有更好、更有效的方法:

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个值.或者，您可以将默认值添加到"值"列表的末尾.