我有两个矩阵,我想在同一个图上的两个子图上有它们对应的两个 3D 图,具有相同的 z 轴。
到目前为止,这是我的代码:
import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d.axes3d import Axes3D
def myplot(matrix1, matrix2):
mymin = np.min(np.array([np.min(matrix1), np.min(matrix2)]))
mymax = np.max(np.array([np.max(matrix1), np.max(matrix2)]))
xsize, ysize = matrix1.shape
x = np.arange(0, ysize, 1)
y = np.arange(0, xsize, 1)
xs, ys = np.meshgrid(x, y)
z1 = matrix1
z2 = matrix2
fig, (ax1, ax2) = plt.subplots(1, 2)
ax1 = Axes3D(fig)
ax1.plot_surface(xs, ys, z1, rstride=1, cstride=1)
ax2 = Axes3D(fig)
ax2.plot_surface(xs, ys, z2, rstride=1, cstride=1)
plt.tight_layout
plt.show()
mat1 = …Run Code Online (Sandbox Code Playgroud)