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

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

f（x，y）=ix^2+jy

我想知道下面的代码是否可以实现这一点，或者是否有更好的实现方法：

def grad_vector(i, j, x, y)

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

---

也许是

u=i*x**2

而不是

u=2*i*y

？对不起，我的意思是

u=2*i*x

和

v=j*y

，理由是u和v是偏导数，请澄清。那么可能是

v=j

而不是

v=j*y

？非常感谢您指出了明显的错误确实应该是

v=j

，那么代码似乎是合理的，并且“做到了”。“是否有更好的方法”取决于许多因素，尤其是在何处/如何使用