Welcome to OGeek Q&A Community for programmer and developer-Open, Learning and Share
Welcome To Ask or Share your Answers For Others

Categories

0 votes
1.3k views
in Technique[技术] by (71.8m points)

python - Plot a 3D Surface of a polyhedron

I'm attempting to use Python and Matplotlib to render a 3D surface of a polyhedron, given by

enter image description here

However my code (shown below) does not seem to draw it correctly. How should this be done instead?

Failed Attempt:

enter image description here

%matplotlib inline

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



delta = 0.1

def x_func(x):
    return abs(x)

def y_func(y):
    return abs(y)

def z_func(z):
    return abs(z)


x = np.arange(-1, 1, delta)
x1 = x_func(x)

y = np.arange(-1, 1, delta)
y1 = y_func(y)

X, Y = meshgrid(x1, y1)

z = np.arange(-1, 1, delta)
Z = z_func(z)

fig = plt.figure()
ax = fig.gca(projection='3d')
ax.set_xlim([-1,1])
ax.set_ylim([-1,1])
ax.set_zlim([-1,1])
surf = ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap=cm.RdBu, linewidth=0.1)
See Question&Answers more detail:os

与恶龙缠斗过久,自身亦成为恶龙;凝视深渊过久,深渊将回以凝视…
Welcome To Ask or Share your Answers For Others

1 Reply

0 votes
by (71.8m points)

Here's one solution:

import mpl_toolkits.mplot3d as a3
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.colors as colors
import scipy as sp
# Vertex data
verts= [    
      (-1, -1, -1), (-1, -1, 1), (-1, 1, 1), (-1, 1, -1),
      (1, -1, -1), (1, -1, 1), (1, 1, 1), (1, 1, -1)
        ]

# Face data
faces = np.array([ 
   [0, 1, 2, 3], [4, 5, 6, 7], [0, 3, 7, 4], [1, 2, 6, 5],     
   [0, 1, 5, 4], [2, 3, 7, 6]
   ])

ax = a3.Axes3D(plt.figure())
ax.dist=30
ax.azim=-140
ax.elev=20[enter image description here][1]
ax.set_xlim([-1,1])
ax.set_ylim([-1,1])
ax.set_zlim([-1,1])

for i in np.arange(len(faces)):
    square=[ verts[faces[i,0]], verts[faces[i,1]], verts[faces[i, 2]], verts[faces[i, 3]]]
    face = a3.art3d.Poly3DCollection([square])
    face.set_color(colors.rgb2hex(sp.rand(3)))
    face.set_edgecolor('k')
    face.set_alpha(0.5)
    ax.add_collection3d(face)

plt.show()

The figure output is this: The surface of a cube


与恶龙缠斗过久,自身亦成为恶龙;凝视深渊过久,深渊将回以凝视…
OGeek|极客中国-欢迎来到极客的世界,一个免费开放的程序员编程交流平台!开放,进步,分享!让技术改变生活,让极客改变未来! Welcome to OGeek Q&A Community for programmer and developer-Open, Learning and Share
Click Here to Ask a Question

...