基于公式的Python梯度向量计算

Question

我认为我走在正确的道路上，但如果我的理解正确，就需要确认比我聪明的人。我想根据公式计算梯度向量：

f（x，y）= i x ^ 2 + j y

我想知道以下代码是否可以解决问题，或者是否有更好的方法：

def grad_vector(i, j, x, y)

    u = 2*i*x
    v = j*y
    return [u, v]