3d在matplotlib的表面

时间:2016-06-10 17:49:29

标签: python matplotlib 3d mplot3d

我已经提到了matplotlib.org和许多线程,但我无法绘制3d表面。我能够绘制一个三维散射图,但是一旦我尝试绘制一个三维表面,就会出现很多不同的错误。我修复错误,但我得到新的错误。与它战斗了2天后,我相信我对matplotlib 3d表面的理解是错误的。

我想绘制SA,FZ,FY

其中SA是X,FZ是Y,FY是Z

 while y<Last_Row:
    if 100 < Temp_AVG_Dict[y] and Temp_AVG_Dict[y] < 140 and abs(Camber[y])<5 and 24.5<Speed[y] and Speed[y]<25.5:
        y = y + 1

        SA[SArow:]=[Run_50_sheet.cell_value( y , SA_Column)]
        SArow = SArow + 1

        FY[FYrow:] = [Run_50_sheet.cell_value( y , FY_Column)]
        FYrow = FYrow + 1

        FZ[FZrow:] = [Run_50_sheet.cell_value( y , FZ_Column)]
        FZrow = FZrow + 1

        Z.insert(SArow,[SA[SArow-1], FZ[FZrow-1], FY[FYrow-1]])

    else:
        y=y+1

Z= np.array(Z)


fig = plt.figure(1)
ax = fig.add_subplot(111, projection='3d')

以下这一行是我认为我出错的地方。 X,Y,Z需要都是2d数组,但我不确定需要输入的值。

我尝试按[SA,FZ],[FZ,FY]和[SA,FY]的顺序制作2d数组,但失败了 我也试过让SA和FY等于1..n,但那也没有用。

ax.plot_surface( SA, FY ,Z)
plt.show()

2d阵列应该由什么值组成? 任何帮助表示赞赏。

1 个答案:

答案 0 :(得分:2)

请查看3d surface的示例代码:http://matplotlib.org/examples/mplot3d/surface3d_demo.html

首先,您必须创建两个常规网格网格X和Y(2d数组)以与plot_surface一起绘制,如下所示:

X = np.arange(-5, 5, 0.25) # X, Y are 1d arrays
Y = np.arange(-5, 5, 0.25)
X, Y = np.meshgrid(X, Y)   # Now X, Y are 2d arrays

尝试制作一维数组FA的网格网格(如果是x),FY(如果是y):

X, Y = np.meshgrid(FA, FY)

然后,创建表示表面值的2d数组Z,即

Z = np.sin(np.sqrt(X**2 + Y**2))

您的FZ数组必须为2d且与X, Y数组具有相同的形状。

情节之后:

ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap=cm.coolwarm,
                       linewidth=0, antialiased=False)

enter image description here

如果网格网格不规则,请查找plot_trisurfhttp://matplotlib.org/examples/mplot3d/trisurf3d_demo.html