在给定数组的情况下迭代方程

Question

所以我想计算给定输入数组的公式的输出 例如

X = np.array([1,2,3,4,5])
variables = np.array([2,5])

def Y(X,variables):
    return = X*(variables[0]-variables[1])

然后是数组 Y 中的输出

我认为 for 循环是一种方式，但我不知道如何。

提前谢谢你

Answer 1

这应该可行：

def Y(X, variables):
    return X*(variables[0]-variables[1])

X = np.array([1,2,3,4,5])
variables = np.array([2,5])

Y = Y(X, variables)
print(Y)

虽然，我不建议你对函数和变量使用相同的名称，所以你可以自己改变它。