python绘制三维点和二维点

来源:互联网 发布:ubuntu 覆盖安装软件 编辑:程序博客网 时间:2024/05/06 09:23

使用matplotlib库绘制二维点
使用mpl_toolkits.mplot3d中的Axes3D绘制三维点

import stringimport matplotlibimport matplotlib.pyplot as pltfrom mpl_toolkits.mplot3d import Axes3Dimport numpy as npif __name__=='__main__':    is_T = False    if is_T:        trajectory = file('./T_groundtruth.txt')        Fig = plt.figure(1)        x = []        y = []        z = []        for line in trajectory:            data = line.split()            if data==[]:                continue            if np.mod(counter,3) == 1:                x.append(string.atof(data[3]))            elif np.mod(counter,3) == 2:                y.append(string.atof(data[3]))            else:                z.append(string.atof(data[3]))        plt.plot( y, z, 'ro-' )        Fig.savefig("trajectory.pdf")    else:        trajectory = file('./sample/trajectory.txt')        #trajectory = file('./livingRoom0_trajectory.txt')        Fig = plt.figure(1)        x = []        y = []        z = []        for line in trajectory:            data = line.split()            x.append( string.atof(data[1]) )            y.append( string.atof(data[2]) )            z.append( string.atof(data[3]) )        ax = Axes3D(Fig)        ax.plot(x,y,z,color='r')        #plt.plot( x, y, 'ro-' )        Fig.savefig("trajectory.pdf")

参考文献
1.https://stackoverflow.com/questions/13570287/image-overlay-in-3d-plot-using-python

原创粉丝点击