Python: отображение интервалов в значения

Я выполняю рефакторинг функции, которая, учитывая серию конечных точек, которые неявно определяют интервалы, проверяет, включено ли число в интервал, а затем возвращает соответствующий (не связанный каким-либо вычислимым образом). Код, который сейчас обрабатывает работу:

if p <= 100:
    return 0
elif p > 100 and p <= 300:
    return 1
elif p > 300 and p <= 500:
    return 2
elif p > 500 and p <= 800:
    return 3
elif p > 800 and p <= 1000:
    return 4
elif p > 1000:
    return 5

Что является IMO довольно ужасно, и не хватает того, что оба интервала и возвращаемые значения жестко закодированы. Любое использование любой структуры данных, конечно, возможно.

6 ответов

Решение
import bisect
bisect.bisect_left([100,300,500,800,1000], p)

вот документы: bisect

Вы можете попробовать это:

def check_mapping(p):
    mapping = [(100, 0), (300, 1), (500, 2)] # Add all your values and returns here

    for check, value in mapping:
        if p <= check:
            return value

print check_mapping(12)
print check_mapping(101)
print check_mapping(303)

производит:

0
1
2

Как всегда в Python, найдутся лучшие способы сделать это.

Это действительно довольно ужасно. Без требования не иметь жесткого кодирования, это должно было быть написано так:

if p <= 100:
    return 0
elif p <= 300:
    return 1
elif p <= 500:
    return 2
elif p <= 800:
    return 3
elif p <= 1000:
    return 4
else:
    return 5

Вот примеры создания функции поиска, как линейного, так и с использованием бинарного поиска, с выполнением требования без жестких кодов, и нескольких проверок работоспособности в двух таблицах:

def make_linear_lookup(keys, values):
    assert sorted(keys) == keys
    assert len(values) == len(keys) + 1
    def f(query):
        return values[sum(1 for key in keys if query > key)]
    return f

import bisect
def make_bisect_lookup(keys, values):
    assert sorted(keys) == keys
    assert len(values) == len(keys) + 1
    def f(query):
        return values[bisect.bisect_left(keys, query)]
    return f

По-другому...

def which(lst, p): 
    return len([1 for el in lst if p > el])

lst = [100, 300, 500, 800, 1000]
which(lst, 2)
which(lst, 101)
which(lst, 1001)
def which_interval(endpoints, number):
    for n, endpoint in enumerate(endpoints):
        if number <= endpoint:
            return n
        previous = endpoint
    return n + 1

Передайте свои конечные точки в виде списка в endpoints, как это:

which_interval([100, 300, 500, 800, 1000], 5)

Редактировать:

Выше приведен линейный поиск. Ответ Гленна Мейнарда будет иметь лучшую производительность, поскольку он использует алгоритм деления пополам.

Попробуйте что-то вроде:

d = {(None,100): 0, 
    (100,200): 1,
    ...
    (1000, None): 5}
value = 300 # example value
for k,v in d.items():
    if (k[0] is None or value > k[0]) and (k[1] is None or value <= k[1]):
        return v
Другие вопросы по тегам