所以我在网上找到了一些使用python3进行优化的代码。修改后,它呈现了这样一个情节
现在我使用的是python2,*
没有被处理。我相信问题是Python迭代,但是当遵循this post中建议的括号技巧时,我没有得到任何结果。完整代码如下:
%matplotlib inline
import matplotlib.pyplot as plt
import pylab as pylab
import autograd.numpy as np
from mpl_toolkits.mplot3d import Axes3D
from matplotlib.colors import LogNorm
from matplotlib import animation
from IPython.display import HTML
from autograd import elementwise_grad, value_and_grad
from scipy.optimize import minimize
from collections import defaultdict
from itertools import izip_longest
from functools import partial
f = lambda x, y: 10*np.cos(1*x) * 15*np.sin(1/2*y) + 150
xmin, xmax, xstep = -4.5, 4.5, .2
ymin, ymax, ystep = -4.5, 4.5, .2
x, y = np.meshgrid(np.arange(xmin, xmax + xstep, xstep), np.arange(ymin, ymax + ystep, ystep))
z = f(x, y)
minima = np.array([np.pi, np.pi])
minima_ = minima.reshape(-1, 1)
fig = plt.figure(figsize=(8, 5))
ax = plt.axes(projection='3d', elev=50, azim=-50)
ax.plot_surface(x, y, z, norm=LogNorm(), rstride=1, cstride=1,
edgecolor='none', alpha=.8, cmap=plt.cm.jet)
ax.plot(*minima_, f(*minima_), 'o', markersize=4, color='w')
ax.set_xlabel('$x$')
ax.set_ylabel('$y$')
ax.set_zlabel('$z$')
ax.set_xlim((xmin, xmax))
ax.set_ylim((ymin, ymax))
plt.show()
以及错误消息:
File "<ipython-input-5-3b03e44c1cac>", line 31
ax.plot(*minima_, f(*minima_), 'o', markersize=4, color='w')
SyntaxError: only named arguments may follow *expression
考虑按索引分配两个最小值。在python 2和python 3中都符合以下要求:
相关问题 更多 >
编程相关推荐