python求函数极值

要在Python中求函数的极值，可以使用不同的方法，其中一种常用的方法是使用牛顿法。牛顿法是一种迭代方法，通过计算函数的导数和二阶导数，来逐步逼近极值点。以下是一个使用牛顿法求解多元函数极值的示例代码：

import scipy.signal as sg
import numpy as np

def get_maxima(values: np.ndarray):
    """找到极大值点"""
    max_index = sg.argrelmax(values)[0]
    return max_index, values[max_index]

def get_minima(values: np.ndarray):
    """找到极小值点"""
    min_index = sg.argrelmin(values)[0]
    return min_index, values[min_index]

if __name__ == '__main__':
    data = np.array([2, 1.5, 1, 1.5, 2, 3, 2, 0, 2, 3, 0])
    print("极值点下标", "极值")
    print(get_minima(data))
    print(get_maxima(data))

这个示例代码使用了Scipy库中的`argrelmax`和`argrelmin`函数来找到函数数据中的极大值和极小值点。你可以将你的函数数据作为输入，并调用相应的函数来获得极值点的下标和值。

另外，如果你想要可视化多元函数的极值点，你可以使用Matplotlib库。以下是一个示例代码，其中使用了Matplotlib中的3D绘图功能来绘制函数曲面和极值点：

import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
import numpy as np

def Fun(x, y):
    return x - y**2 - 2*x**2 - 2*x*y

def PxFun(x, y):
    return 1 - 4*x - 2*y

def PyFun(x, y):
    return -1 - 2*x - 2*y

fig = plt.figure()
ax = Axes3D(fig)

X, Y = np.mgrid[-2:2:40j, -2:2:40j]
Z = Fun(X, Y)

ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap="rainbow")
ax.set_xlabel('x')
ax.set_ylabel('y')
ax.set_zlabel('z')

step = 0.0008
x = 0
y = 0
tag_x = [x]
tag_y = [y]
tag_z = [Fun(x, y)]

new_x = x
new_y = y
Over = False

while Over == False:
    new_x -= step * PxFun(x, y)
    new_y -= step * PyFun(x, y)
    
    if Fun(x, y) - Fun(new_x, new_y) < 7e-9:
        Over = True
    
    x = new_x
    y = new_y
    
    tag_x.append(x)
    tag_y.append(y)
    tag_z.append(Fun(x, y))

ax.plot(tag_x, tag_y, tag_z, 'r.')
plt.title('(x,y)~(' + str(x) + "," + str(y) + ')')
plt.show()

这个示例代码中定义了一个多元函数`Fun`，以及其对x和y的偏导数函数`PxFun`和`PyFun`。然后使用取样点的坐标和函数值创建了一个3D图像，并使用梯度下降方法找到了该函数的极值点。最后，使用红色的点将极值点在图像中标出。

希望这些示例代码对你有所帮助，可以让你在Python中求解函数的极值。