python - 函数返回返回 int 或 int 列表。将第一个赋值给一个变量

Question

假设我们有

def f_1():
    return (3, 4)

def f_2():
    return 2

给出了函数。这不是我可以修改的代码。我们知道它们返回一个整数或它们的序列。

我想将返回值分配给一个变量，如果返回值是整数的序列，则变量应该只取第一个整数。

Python 中是否有内置的句法设备允许我执行此操作？

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

注意:假设我们不知道哪些函数返回序列，哪些只返回一个数字。

Answer 1

Is there a build-in syntactic device

没有。通常的做法是ask对于 forgivness :

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