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

时间:2014-04-15 16:15:58

标签: python function return iterable

我需要使用包中的函数来返回单个对象或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是包的一部分,我当然无法修改它。

是否有一种优雅的方式(例如,无需检查输出是否可迭代)?

0 个答案:

没有答案