Estoy usando una biblioteca de python con una función que devuelve resultados como una lista. Dependiendo de la configuración del experimento, esta lista se compone de uno o dos elementos. En este caso, una forma sencilla de asignar la salida de la función es la siguiente:
bbox, segm = None, None results = test_model() # returns a list with either one or two elements bbox = results[0] if len(results) > 1: segm = results[1] # [some other code here] if segm is not None: plot(segm) Sin embargo, esto parece ser bastante detallado porque primero debemos inicializar tanto bbox como segm en None y luego evaluar if len(results) > 1 . ¿Hay alguna forma pitónica de evitar esto? Idealmente, algo como esto sería muy bueno:
bbox, segm = test_model() # returns a list with either one or two elements if segm is not None: plot(segm)Puede usar la coincidencia de patrones estructurales , disponible en Python 3.10:
match test_model(): case [bbox, segm]: plot(segm) case bbox: pass Sin embargo, si test_module siempre devuelve una tupla, puede usar el desempaquetado:
bbox, *segm = test_model() Ahora, si text_model devuelve un valor, segm será una lista vacía ( [] ). Sin embargo, si devuelve dos valores, segm contendrá un solo elemento. Por lo tanto, su código completo puede convertirse en:
bbox, *segm = test_model() if segm: plot(segm[0])Si está en Python 3.10, vaya con la respuesta de Ajax1234 .
De lo contrario, tu prioridad es escribir algo que sea fácil de leer en lugar de hacer algo elegante. Asignar a una tupla como esta funcionaría:
try: bbox, segment = results except ValueError: bbox, segment = results[0], NoneEsto tiene un montón de ventajas:
Cuando se trabaja con excepciones, tiene sentido usar la excepción para el caso que realmente es excepcional. Entonces, si espera muchos más resultados con un elemento de la lista, puede escribirlo al revés:
try: bbox, = results segment = None except ValueError: bbox, segment = resultsPuede extender la list de return un elemento adicional y luego cortar los 2 elementos necesarios.
def foo(boolean): if boolean: return [True, True] return [False] bar, baz = (foo(False) + [None])[:2] print(bar, baz) # False None bar, baz = (foo(True) + [None])[:2] print(bar, baz) # True True La forma pitónica sería * desempaquetar como se mencionó @ Ajax1234, pero esta es una alternativa.
Si tiene el control de la definición de la función, también puede agregar un decorador.
def returns_two(func): def wrapper(*args): return (func(*args) + [None])[:2] return wrapper @returns_two def foo(boolean): if boolean: return [True, True] return [False] bar, baz = foo(False) print(bar, baz) # False None bar, baz = foo(True) print(bar, baz) # True True