Как сделать так, чтобы мой график поверхности отображался с помощью Axes3D?

Question

Я пытаюсь создать поверхностный график, используя Python Matplotlib.Я прочитал документацию, пытаясь выяснить, где мой код был неправильным или я что-то пропустил, но у меня возникли проблемы.

Код, который я написал, -

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

def computeCost(X, y, theta):
    m = len(y)
    predictions = np.dot(X, theta)
    squareErros = (predictions - y) ** 2
    J = (1 / (2 * m)) * sum(squareErrors)

    return J

data = np.loadtxt("./data1.txt", delimiter=',')
X = data[:, 0].reshape(-1, 1)
y = data[:, 1].reshape(-1, 1)
m = len(y)
X = np.concatenate((np.ones((m, 1)), X), axis=1)

theta0_vals = np.linspace(-10, 10, 100) # size (100,)
theta1_vals = np.linspace(-1, 4, 100) # size (100,)
J_vals = np.zeros((len(theta0_vals), len(theta1_vals)))

for i in range(len(x_values)):
    for j in range(len(y_values)):
        t = np.array([theta0_vals[i], theta1_vals[j]]).reshape(-1, 1)
        J_vals[i][j] = computeCost(X, y, t) # size (100, 100)

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
surf = ax.plot_surface(theta0_vals, theta1_vals, J_vals)

plt.show()

Когда я вызываю plt.show(), я не получаю вывод.Поверхностный сюжет, который я ожидаю увидеть, похож на этот:

Кто-нибудь будет достаточно любезен, чтобы сообщить мне, где я используюбиблиотека поверхностного участка пошла не так?Спасибо.

РЕДАКТИРОВАТЬ

Я попытался запустить предоставленный демонстрационный код здесь и этоработает отлично.Вот код для этого:

from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt
from matplotlib import cm
from matplotlib.ticker import LinearLocator, FormatStrFormatter
import numpy as np


fig = plt.figure()
ax = fig.gca(projection='3d')

# Make data.
X = np.arange(-5, 5, 0.25)
Y = np.arange(-5, 5, 0.25)
X, Y = np.meshgrid(X, Y)
R = np.sqrt(X**2 + Y**2)
Z = np.sin(R)

# Plot the surface.
surf = ax.plot_surface(X, Y, Z, cmap=cm.coolwarm,
                       linewidth=0, antialiased=False)

# Customize the z axis.
ax.set_zlim(-1.01, 1.01)
ax.zaxis.set_major_locator(LinearLocator(10))
ax.zaxis.set_major_formatter(FormatStrFormatter('%.02f'))

# Add a color bar which maps values to colors.
fig.colorbar(surf, shrink=0.5, aspect=5)

plt.show()

Answer 1

Я думаю, что я понял проблему, изменив пару последних строк кода с

ax = fig.add_subplot(111, projection='3d')
surf = ax.plot_surface(theta0_vals, theta1_vals, J_vals)

на

ax = plt.axes(projection='3d')
surf = ax.plot_surface(theta0_vals, theta1_vals, J_vals, rstride=1, cstride=1, cmap='viridis', edgecolor='none')

Внесение этого изменения дает мне поверхностный графиктакой, что:

Ссылка, которая дала мне ссылку на это, была this .