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