用pyplot绘制2d数组

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 examples

中看到更复杂，更完整的曲面图示例