Python SciPy optimize.rosen用法及代碼示例

本文簡要介紹 python 語言中 scipy.optimize.rosen 的用法。

用法:

scipy.optimize.rosen(x)#

羅森布洛克函數。

計算的函數是：

sum(100.0*(x[1:] - x[:-1]**2.0)**2.0 + (1 - x[:-1])**2.0)

參數 ：：

x： array_like

要計算 Rosenbrock 函數的點的一維數組。

返回 ：：

f： 浮點數

Rosenbrock 函數的值。

例子：

>>> import numpy as np
>>> from scipy.optimize import rosen
>>> X = 0.1 * np.arange(10)
>>> rosen(X)
76.56

對於高維輸入rosen 廣播。在以下示例中，我們使用它來繪製 2D 風景。請注意，rosen_hess 不會以這種方式廣播。

>>> import matplotlib.pyplot as plt
>>> from mpl_toolkits.mplot3d import Axes3D
>>> x = np.linspace(-1, 1, 50)
>>> X, Y = np.meshgrid(x, x)
>>> ax = plt.subplot(111, projection='3d')
>>> ax.plot_surface(X, Y, rosen([X, Y]))
>>> plt.show()