如何使用8点形成平面?

时间:2019-06-12 12:23:59

标签: python arrays 3d surface plane

我已经绘制了我的8个角点,其长方体中心如图8_corner_points_with center所示。

我已经尝试了散点图,但是现在我想要连接这8个点的曲面图。

当我尝试使用表面图时,我无法解决这个问题。您能提出任何解决方案的建议

l = 0.3 
w = 0.4
h = 0.1
center = 
 [2.10737, -0.100085, 0.716869]
F= 
   [[array([[1.]]) array([[-0.001]]) array([[-0.017]])]
 [array([[0.]]) array([[-0.999]]) array([[0.037]])]
 [array([[0.017]]) array([[0.037]]) array([[0.999]])]]


def cuboid(center, size):
    ox, oy, oz = center
    l, w, h = size

    ax = fig.gca(projection='3d') ##plot the project cuboid

    X=[ox-l/2,ox-l/2,ox-l/2,ox-l/2,ox+l/2,ox+l/2,ox+l/2,ox+l/2]
    Y=[oy+w/2,oy-w/2,oy-w/2,oy+w/2,oy+w/2,oy-w/2,oy-w/2,oy+w/2]
    Z=[oz-h/2,oz-h/2,oz+h/2,oz+h/2,oz+h/2,oz+h/2,oz-h/2,oz-h/2]

    X_new = ([])
    Y_new = ([])
    Z_new = ([])

    for i in range(0,8):

        c=np.matrix([[X[i]],
                    [Y[i]],
                    [Z[i]]])
        u=F*c

        X_new = np.append(X_new, u.item(0))
        Y_new = np.append(Y_new, u.item(1))
        Z_new = np.append(Z_new, u.item(2))

    ax.scatter(X_new,Y_new,Z_new,c='darkred',marker='o')  #the  plot of points after rotated
    ax.scatter(ox,oy,oz,c='crimson',marker='o')  #the previous plot of points before rotated



    ## Add title 
    plt.title('Plot_for_PSM', fontsize=20)
    plt.gca().invert_yaxis()

    ##labelling the axes
    ax.set_xlabel('X')
    ax.set_ylabel('Y')
    ax.set_zlabel('Z')


1 个答案:

答案 0 :(得分:1)

enter image description here

这里是一个解决方案。

###Added import
import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
from mpl_toolkits.mplot3d.art3d import Poly3DCollection, Line3DCollection

###Addded size and center in this format in order to manipulate them easily
size = [0.3, 0.4, 0.1]
center = [2.10737, -0.100085, 0.716869]

###This numpy vector will be used to store the position of the sides
side = np.zeros((8,3))

###Just re-ordered your matrix in some np.arrays
F = [[np.array([1., -0.001, -0.017])],
    [np.array([0., -0.999, 0.037])],
    [np.array([0.017, 0.037, 0.999])]]

def cuboid(center, size):
    ox, oy, oz = center
    l, w, h = size

    ###Added the fig in order to be able to plot it later
    fig = plt.figure()
    ax = fig.gca(projection='3d') ##plot the project cuboid

    X=[ox-l/2,ox-l/2,ox-l/2,ox-l/2,ox+l/2,ox+l/2,ox+l/2,ox+l/2]
    Y=[oy+w/2,oy-w/2,oy-w/2,oy+w/2,oy+w/2,oy-w/2,oy-w/2,oy+w/2]
    Z=[oz-h/2,oz-h/2,oz+h/2,oz+h/2,oz+h/2,oz+h/2,oz-h/2,oz-h/2]

    X_new = ([])
    Y_new = ([])
    Z_new = ([])

    for i in range(0,8):

        c=np.matrix([[X[i]],
                    [Y[i]],
                    [Z[i]]])
        u=F*c

        X_new = np.append(X_new, u.item(0))
        Y_new = np.append(Y_new, u.item(1))
        Z_new = np.append(Z_new, u.item(2))

        ###Doing a dot product between F and c like you did earlier but using np.dot as we're now working with Numpy format
        side[i,:] = np.dot(F, c)

    ###Storing the position of every points
    sides = [[side[0],side[1],side[2],side[3]],
            [side[4],side[5],side[6],side[7]], 
            [side[0],side[1],side[4],side[5]], 
            [side[2],side[3],side[4],side[5]], 
            [side[1],side[2],side[5],side[6]],
            [side[4],side[7],side[0],side[3]]]


    ###Scatter plot 
    ax.scatter(X_new,Y_new,Z_new,c='darkred',marker='o')  #the  plot of points after rotated
    ax.scatter(ox,oy,oz,c='crimson',marker='o')  #the previous plot of points before rotated



    ### Add title 
    plt.title('Plot_for_PSM', fontsize=20)
    plt.gca().invert_yaxis()

    ##labelling the axes
    ax.set_xlabel('X')
    ax.set_ylabel('Y')
    ax.set_zlabel('Z')

    ###This draw the plane sides as you wanted
    ax.add_collection3d(Poly3DCollection(sides, facecolors='blue', linewidths=1, edgecolors='r', alpha=.25))


cuboid(center, size)
###Mandatory to plot the cube
plt.show()

它使用Poly3DCollection, Line3DCollection中的mpl_toolkits绘制6个平面正方形,代表立方体的侧面。

第一步是找到每侧的4个坐标。然后,您需要使用Poly3DCollection进行绘制。