cómo comprobar variable con 2 valores posibles pitón

Question

tengo un s variable que contiene una cadena de una letra

s = 'a' 

Dependiendo del valor de esa variable, quiero volver las cosas diferentes. Hasta ahora estoy haciendo algo similar a esto:

if s == 'a' or s == 'b': 
    return 1 
elif s == 'c' or s == 'd': 
    return 2 
else: 
    return 3 

¿Existe alguna forma mejor de escribir esto? una forma más pythonic? ¿O es este el más eficiente?

Anteriormente, he tenido incorrectamente algo como esto:

if s == 'a' or 'b': 
    ... 

Obviamente eso no funciona y era bastante tonto de mí.

sé de asignación condicional y han intentado esto:

return 1 if s == 'a' or s == 'b' ... 

Creo que mi pregunta es específicamente para ¿hay alguna manera se puede comparar una variable de dos valores sin tener que escribir 'algo == O algo algo == algo '

Answer 1

if s in ('a', 'b'): 
    return 1 
elif s in ('c', 'd'): 
    return 2 
else: 
    return 3 

Answer 2

Si solo devuelve valores fijos, un diccionario es probablemente el mejor enfoque.

Answer 3

if s in 'ab': 
    return 1 
elif s in 'cd': 
    return 2 
else: 
    return 3 

Answer 4

d = {'a':1, 'b':1, 'c':2, 'd':2} 
return d.get(s, 3) 

Answer 5

Tal vez poco más autodocumentado de usar si otra cosa:

d = {'a':1, 'b':1, 'c':2, 'd':2} ## good choice is to replace case with dict when possible 
return d[s] if s in d else 3 

También es posible aplicar la popular primera respuesta con si otra cosa:

return (1 if s in ('a', 'b') else (2 if s in ('c','d') else 3)) 

Answer 6

return 1 if (x in 'ab') else 2 if (x in 'cd') else 3