un modo pietonico per scrivere una funzione constrain()

Question

Qual è il modo migliore di scrivere una funzione di vincolo? o c'è già una funzione python integrata che fa questo?

Opzione 1:

def constrain(val, min_val, max_val): 

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

Opzione 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

Non so se questo è il più "divinatorio", ma è possibile utilizzare built-in min() e max() come questo:

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

Answer 2

Nel caso in cui sia necessario eseguire questa operazione per molti numeri (array completi di essi), è probabile che si stia utilizzando Numpy, che ha una funzione integrata clip. Per semplici programmi Python, vai su Delgan answer.

Answer 3

Se è possibile elaborare una serie di valori in un momento, si potrebbe provare una lista di comprensione:

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] 

o anche 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])