使用表面和线框使用Matplotlib建模小行星

时间:2012-11-27 16:47:29

标签: python matplotlib plot wireframe geometry-surface

我正在尝试使用plot_surfaceplot_wireframe对小行星进行建模。我对小行星表面上的点有x y和z值。线框精确到小行星的形状,但表面图不适合线框。如何获得适合线框的曲面图或如何使用线框获得3d实体模型?这是我的模型代码:

from mpl_toolkits.mplot3d import axes3d
import matplotlib.pyplot as plt
import numpy as np
from matplotlib import cm
from matplotlib.mlab import griddata

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
data = np.genfromtxt('data.txt')
x = data[:,0]
y = data[:,1]
z = data[:,2]

ax.plot_wireframe(x, y, z, rstride=1, cstride=1, alpha=1)

xi = np.linspace(min(x), max(x))
yi = np.linspace(min(y), max(y))

X, Y = np.meshgrid(xi, yi)
Z = griddata(x, y, z, xi, yi)


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

plt.show()

数据采用这种格式,尽管原始文件中有更多行:

-1.7738946051191869E-002   4.3461451610545973E-002   1.3393057231408241     
-0.29733561550902488       0.32305812106837900        1.3393057231408241     
-0.29733561550902488       0.16510132228266330        1.3548631099230350     
-0.21872587865015569        2.4170900455101410E-002   1.3610011616437809     
1.4452975249810950E-002 -0.20900795344486520        1.3610011616437809    
1.5732454381265970E-002 -0.20900795344486520        1.3608751439485580    
-0.34501536374240321       0.51320241386595655        1.3158820995876130     
-0.40193014435941982       0.45628763324893978        1.3158820995876130     
-0.42505849480150409       0.28183419537116011        1.3307863198123011     
-0.18994178462386799      -0.19294290416565860        1.3424523041534830     
1.4452975249810939E-002 -0.39733766403933751        1.3424523041534830     
5.8021940902131752E-002 -0.57108837516584876        1.3210481842104100     
9.3746267961881152E-002 -0.61017602710257668        1.3136798474111200     
0.26609469681891229      -0.43782759824554562        1.3136798474111200     
  0.17938460413447810       0.39179924148155021        1.2357401964919650     
   8.9613011902522258E-002  0.42818009222325598        1.2584008460875080     
  0.33671539027096409      -0.47165177581327772        1.2965073126705291     
  0.53703772594296528      -0.47165177581327777        1.2357401964919561     
 -0.19242375014122229       0.71021685426700043        1.2584008460875080     
 -0.34501536374240321       0.66763766324752027        1.2904902860951690 

希望你能帮忙

1 个答案:

答案 0 :(得分:2)

你能提供一份正在发生的事情的图片吗?我猜你正在得到一个图像,小行星的表面似乎在各处跳跃。那是对的吗?如果是这样,可能是绘图仪不了解点的顺序造成的。

如果我们有一组包含单位圆的所有点的点,那么我们期望绘制这些点以创建单位圆。但是,如果您决定将每个点连接到另外两个点,则它看起来不一定像圆圈。如果(出于某种原因)你将一个点连接到圆圈另一侧的另一个点并继续这样做,直到每个点连接到另外两个点,它可能看起来也可能看起来不像一个圆圈,因为每个点都不是必然连接到相邻的点。

你的小行星也是如此。你需要提出一些方案,以便绘图仪知道如何连接点,否则你将继续遇到同样的问题。

以下圆圈示例应说明我的观点:

import math
import matplotlib.pylab as plt
import random

thetaList = range(360)
random.shuffle(thetaList)
degToRad = lambda x: float(x) * math.pi / float(180)
x = [math.cos(degToRad(theta)) for theta in thetaList]
y = [math.sin(degToRad(theta)) for theta in thetaList]       

#plot the cirlce
plt.plot(x,y)
plt.show()