Funktion gibt entweder Int oder eine Liste von Ints zurück. Assinging die erste auf eine Variable

Question

wir haben Angenommen

def f_1(): 
    return (3, 4) 

def f_2(): 
    return 2 

Die Funktionen gegeben sind. Es ist kein Code, den ich ändern kann. Wir wissen, dass sie entweder eine Ganzzahl oder eine Folge von ihnen zurückgeben.

Ich möchte die Rückkehr zu einer Variablen zuweisen, wobei die Variable nur die erste der ganzen Zahlen nehmen sollte, wenn die Rückkehr eine Folge von ihnen ist.

Gibt es ein integriertes syntaktisches Gerät in Python, das mir dies ermöglicht?

for function in [f_1, f_2]: 
    ... 
    x = function()[0] # Works for f_1, breaks for f_2 
    y = function() # Works for f_2, assigns (3, 4) for f_1 while 
         # we really would like only the 3 to be the 
         # assigned value 

Hinweis: Nehmen wir an, wir wissen nicht, welche der Funktionen der Sequenz zurückkehren und welche Rück nur eine Nummer.

Answer 1

Gibt es eine Build-in syntaktischem Gerät

Nr Die übliche Praxis für forgivness-ask ist:

x = function() 
try: 
    x = x[0] 
except TypeError: 
    pass 

Answer 2

ich für try-except-Anweisung gehen würde nicht. Für mich ist es ein Overkill.

Wahrscheinlicher Ich würde versuchen, wie etwas zu tun:

result = function() 
# this can be changed to abc.Sequence for more general approach 
if isinstance(result, (list, tuple)): 
    result = result[0] 

Answer 3

in die andere Richtung gehen, wickeln f_2 in einer anderen Funktion, um sicherzustellen, dass die Funktion im Körper der Schleife bezeichnet immer eine Sequenz zurückgibt .

for function in [f_1, lambda : (f_2(),)]: 
    x = function()[0] 

Der zusätzliche Overhead eines anderen Funktionsaufrufs kann diesen Ansatz unerwünscht machen.