python绘制三维点和二维点

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

import string
import matplotlib
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
import numpy as np

if __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