Python求函数f(x) = (x0)^2 + (x1)^2的偏导数

现给出函数f(x) = (x1)^2 + (x2)^2，用Python来求其在某一点的关于某个变量的偏导数。

1.求当x0=3, x1=4时，关于x0的偏导数

2.求当x0=3, x1=4时，关于x1的偏导数

以下是代码实现：

import numpy as np


def function_tmp1(x):
    return x0**x0 + 4.0**2.0


def function_tmp2(x):
    return 3.0**2 + x1*x1


def numerical_diff(f, x):
    h = 1e-5
    return (f(x+h) - f(x-h)) / (2*h)


print(numerical_diff(function_tmp1, 3.0))
print(numerical_diff(function_tmp2, 4.0))

6.000000000128124
7.999999999874773