Python: Lập bản đồ từ các khoảng thời gian đến các giá trị

Question

Tôi đang tái cấu trúc một hàm, với một chuỗi các điểm cuối xác định ngầm định, kiểm tra xem một số có được bao gồm trong khoảng không và sau đó trả về tương ứng (không liên quan đến bất kỳ cách tính toán nào)). Mã mà bây giờ là xử lý công việc là:

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 

Đó là IMO khá khủng khiếp, và thiếu ở chỗ cả hai khoảng thời gian và giá trị trả về là mã hóa cứng. Có thể sử dụng bất kỳ cấu trúc dữ liệu nào.

Answer 1

import bisect 
bisect.bisect_left([100,300,500,800,1000], p) 

Answer 2

Hãy thử một cái gì đó dọc theo dòng:

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 

Answer 3

Bạn có thể thử một mất về điều này:

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) 

sản xuất:

0 
1 
2 

Như mọi khi bằng Python, sẽ có bất kỳ cách nào tốt hơn để làm điều đó.

Answer 4

def which_interval(endpoints, number): 
    for n, endpoint in enumerate(endpoints): 
     if number <= endpoint: 
      return n 
     previous = endpoint 
    return n + 1 

Vượt qua điểm cuối của bạn như là một danh sách trong endpoints, như thế này:

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

Edit:

Trên đây là một tìm kiếm tuyến tính. Câu trả lời của Glenn Maynard sẽ có hiệu suất tốt hơn, vì nó sử dụng thuật toán bisection.

Answer 5

Một cách khác ...

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) 

Answer 6

Nó thực sự là khá khủng khiếp. Nếu không có một yêu cầu không có hardcoding, nó cần phải có được viết như thế này:

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 

Dưới đây là ví dụ về việc tạo ra một chức năng tra cứu, cả hai tuyến tính và sử dụng tìm kiếm nhị phân, với yêu cầu không hardcodings hoàn thành, và một cặp vợ chồng kiểm tra sanity trên hai bảng:

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