使用陣列繪製3D表面

Question

Z=np.array([[10.,12.,12.,5.], 
     [10.,0.,0.,5.], 
     [10.,0.,0.,5.], 
     [10.,20.,20.,20.]]) 
X = np.arange(0, 4, 1) 
Y = np.arange(0, 4, 1) 

我有一個2D 4x4陣列。我想製作一個三維繪圖，x軸和y軸具有從0到4的離散整數值。有人可以幫助我嗎？

Answer 1

首先需要讓你X的二維數組，Y向量：

import numpy as np 
X2D,Y2D = np.meshgrid(X,Y) 

那麼你可以使用曲面圖（或線框）：

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

fig = plt.figure() 
ax = Axes3D(fig) 

ax.plot_surface(X2D,Y2D, Z) 

軸將只能是0到3如果你只有4分（你需要5從0到4）