Построение 2d массива с pyplot

Question

У меня есть два 1d массива x и t, которые составляют сетку, и 2d массив z для этой сетки.
z[i][j] означает z(t[i], x[j]).Как построить эту поверхность с помощью pyplot ?

Answer 1

Начиная с массивов 1d, создайте сетку 2d и используйте plt.surface().

# Make sure all necessary libraries are loaded
from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt
import numpy as np

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

# create meshgrid
T, X = np.meshgrid(t, x)

# plot
ax.plot_surface(T, X, Z)

Более сложный и полный пример поверхностного графика можно увидеть в примерах matplotlib