时间戳

首页 > 解决方案 > 如何有效地检查python中的连续范围

python - 如何有效地检查python中的连续范围

问题描述

根据范围分配等级：

def getGrade(size):
    grade =''
    if size <= 32:
        grade = 'p4'
    elif size > 32 and size <=64:
        grade = 'p6'
    elif size > 64 and size <= 128:
        grade = 'p10'
    elif size > 128 and size <= 256:
        grade = 'p15'
    elif size > 256 and size <=512:
        grade = 'p20'
    elif size > 512 and size <= 1024:
        grade = 'p30'
    elif size > 1024 and size <= 2048:
        grade = 'p40'
    ......

问题是需要再添加 20 个检查，所以有没有比这种方法做得更好的方法。

标签： pythonpython-3.xrange

解决方案

由于范围是连续的，您可以避免重复下限。

将所有范围放在元组中可以节省一些输入（如果第一个范围没有下降到负无穷大，请考虑(0, None)在所有其他范围之前添加元组：

def getGrade(size):
    grades = (
         (32, 'p4'),
         (64, 'p6'),
        (128, 'p10'),
        ...
    )

    for maxVal, grade in grades:
        if size <= maxVal:
            return grade

测试：

>>> getGrade(45)
'p6'
>>> getGrade(100)
'p10'

效率：

如果grades列表真的很长，您可以获得比扫描每个项目更好的运行时间。由于列表已排序，因此您可以bisect通过替换 for 循环来使用 ,：

    for maxVal, grade in grades:
        if size <= maxVal:
            return grade

和：

    index = bisect.bisect(grades, (size, ))
    if index < len(grades):
        return grades[index][1]

步骤数（在最坏的情况下）从 N（的长度grades）减少到 log 2 (N)。

推荐阅读