我需要使用包中的函数来返回单个对象或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
是包的一部分,我当然无法修改它。
是否有一种优雅的方式(例如,无需检查输出是否可迭代)?