Как найти все локальные минимумы в 3D Python Plot? - PullRequest
0 голосов
/ 10 декабря 2018

Я хочу показать локальные оптимумы трехмерного невыпуклого графика.Это код, который может отображать сюжет, но не локальный и глобальный оптимум:

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()
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...