cách nhiệt đới để viết hàm constrain()

Question

Cách tốt nhất để viết hàm constrain là gì? hoặc là đã có một hàm python dựng sẵn thực hiện điều này?

Lựa chọn 1:

def constrain(val, min_val, max_val): 

    if val < min_val: return min_val 
    if val > max_val: return max_val 
    return val 

Phương án 2:

def constrain(val, min_val, max_val): 

    if val < min_val: 
     val = min_val 

    elif val > max_val: 
     val = max_val 

    return val 

Answer 1

Tôi không biết nếu điều này là nhiều hơn "pythonic", nhưng bạn có thể sử dụng built-in min() và max() như sau:

def constrain(val, min_val, max_val): 
    return min(max_val, max(min_val, val)) 

Answer 2

Trong trường hợp bạn cần thực hiện điều này cho nhiều số (mảng đầy đủ), có thể bạn đang sử dụng Numpy, có chức năng được xây dựng trong clip. Đối với các chương trình Python đơn giản, hãy truy cập với answer của Delgan.

Answer 3

Nếu bạn có thể xử lý một loạt các giá trị cùng một lúc, bạn có thể thử một danh sách hiểu:

a = [1,1,5,1,1] 
b = [7,2,8,5,3] 
c = [3,3,3,3,3] 
[min(y,max(x,z)) for x,y,z in zip(a, b, c)] 

[3, 2, 5, 3, 3] 

hoặc thậm chí NumPy:

import numpy as np 

a = np.array(a) 
b = np.array(b) 
c = np.array(c) 

np.minimum(b, np.maximum(a, c)) 
np.minimum(b, np.maximum(a, 3)) # just use 3 if they are all the same 

c.clip(a, b) # or just use NumPy's clip method 
np.clip(c, a, b) 

array([3, 2, 5, 3, 3]) 
array([3, 2, 5, 3, 3]) 
array([3, 2, 5, 3, 3]) 
array([3, 2, 5, 3, 3])