Python：读取具有可变返回类型的函数的输出

Question

我需要使用包中的函数来返回单个对象或numpy对象数组。我想在numpy数组中读取函数的输出，而不管输出的类型。

这样做的原因是我希望能够迭代输出。所以实际上它不一定是一个numpy数组，我只想要一个可迭代的对象。

例如：

import numpy as np

def f(returnArray) :
  if returnArray :
    return np.arange(2)
  else :
    return 1

foo1 = f(True)
foo2 = f(False)

print foo1, foo2

输出

[0 1] 1

但我希望foo2可以迭代。由于f是包的一部分，我当然无法修改它。

是否有一种优雅的方式（例如，无需检查输出是否可迭代）？