我正在使用Scipy.Optimize.fmin來查找函數的最大值。輸出的形式是numpy.ndarray,其中包含有關進程的其他信息。我只需要將x值作爲浮點數返回。如何從numpy.ndarray中提取值
def f(x):
"""returns the value of f(x) with the input value x"""
import math
f = math.exp(-x ** 2.0)/(1.0 + x ** 2.0) + \
2.0 * (math.cos(x) ** 2.0)/(1.0 + (x - 4.0) ** 2.0)
return f
def find_max_f():
"""returns the x for which f(x) takes the maximum value"""
import scipy.optimize as o
m = o.fmin(lambda x: -f(x), 0)
return m
這是它返回:
>>> find_max_f()
Optimization terminated successfully.
Current function value: -1.118012
Iterations: 12
Function evaluations: 24
array([ 0.0131875])
我只需要標記陣列括號內的最終數目
'fmin()'的返回類型是什麼?如果它是可迭代的,你可以使用索引來獲取它。 – IanAuld