0

我有要绘制的常规网格。我使用了plot_surface，但结果有点奇怪。可以帮忙画图吗？我试图改变轴的限制，但它没有帮助。也许你可以建议更好的方法来绘制点作为表面。谢谢！

from mpl_toolkits.mplot3d import Axes3D
from matplotlib import cm
from matplotlib.ticker import LinearLocator, FormatStrFormatter
import matplotlib.pyplot as plt
import numpy as np
from array import *

x = np.array([0, 0, 0, 400, 400, 400, 800, 800, 800])
y = np.array([0, 400, 800, 0, 400, 800, 0, 400, 800])
z = np.array([10.0, 10.0, 10.0, 10.0, 206.8, 10.0, 10.0, 10.0, 10.0])

#print len(x), len(y), len(z)

zdict = {}

for i in range (0, len(x)):
    #print "(", x[i], ", ", y[i], "): ", z[i], ", "
    zdict[(x[i], y[i])] = z[i]

print zdict

def zfunc(x, y):
    return zdict[(x, y)]

X, Y = np.meshgrid(x, y)

print X, Y

#need 2d array of Z

Z = [[zfunc(X[i][j], Y[i][j]) for i in range(0, len(X))] for j in range(0,    len(X[0]))]

print Z

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

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

ax.zaxis.set_major_locator(LinearLocator(10))
ax.zaxis.set_major_formatter(FormatStrFormatter('%.02f'))

fig.colorbar(surf, shrink=0.5, aspect=5)

plt.show()
4

1 回答 1

0

您的网格是错误的：它包含许多重复x的y值。试试这个：

In [7]: X,Y = meshgrid(unique(x), unique(y))

In [8]: X
Out[8]: 
array([[  0, 400, 800],
       [  0, 400, 800],
       [  0, 400, 800]])

In [9]: Y
Out[9]: 
array([[  0,   0,   0],
       [400, 400, 400],
       [800, 800, 800]])

In [10]: Z = z.reshape((3,3))

In [11]: fig = plt.figure()
    ...: ax = fig.gca(projection='3d')
    ...: surf = ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap=cm.coolwarm, linewidth=0)
于 2015-05-24T10:41:23.520 回答