我想展示3D非凸图的局部最优 . 这是可以显示情节的代码,但不是本地和全局最优:

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

#Input data is something like this[x,y,z]:
TT = [[2,1,450], [3,1,678],[0,0,890]

def fun(x, y):
    for r in range(len(TT)):
        if x == TT[r][0] and y == TT[r][1]:
            return TT[r][2]

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
x = y = np.arange(-2.0, 3.0, 1.0)
X, Y = np.meshgrid(x, y)
zs = np.array([fun(x,y) for x,y in zip(np.ravel(X), np.ravel(Y))])
Z = zs.reshape(X.shape)

ax.plot_surface(X, Y, Z ,cmap=cm.coolwarm,)

plt.show()