我有一个依赖于几个变量的函数,比如说y=f(x1,x2,x3,x4)
. 如果每个变量都是一个简单的数字,那么结果应该是一个普通的数字。如果其中一个变量是一个数组,我需要结果也是一个数组。依此类推:如果两个变量是数组,我需要结果是一个二维数组。等等。
例子:
def f(x1,x2,x3,x4):
y=x1*x2/(x3+x4)
return y
x1=1.0
x2=2.0
x3=3.0
x4=4.0
f(x1,x2,x3,x4)
# should give 2.0/7.0 = 0.2857...
x3=array([1.0,2.0,3.0,4.0,5.0])
f(x1,x2,x3,x4)
# should give a one-dimensional array with shape (5,)
x4=array([10.0,20.0,30.0,40.0,50.0,60.0,70.0])
f(x1,x2,x3,x4)
# should give a two-dimensional array with shape (5,7)
怎么做?(为了让我的程序的非 Python 读者尽可能清楚?)