我正在编写一个函数,希望当输入为单个元素时,函数返回一个元素;当输入为数组时,函数返回一个由输出元素组成的数组,且每个输出元素与输入数组中相应位置的元素相关联。以下是一个示例:
import numpy as np
def f(a):
if a<5:
print a;
f(np.arange(11))
这段代码返回了错误:
if a<5:
ValueError: 数组的真值有多个,无法确定。请使用 a.any() 或 a.all()
我期望输出结果为:
0
1
2
3
4
我该如何使它按我所解释的方式工作,因为我相信许多Python函数都是这样工作的呢?谢谢。
if a<5
做什么? - Daniel Roseman