Skip to content

Instantly share code, notes, and snippets.

@neale

neale/plot_trajectory.py

Last active June 12, 2023 16:52
Download ZIP

Revisions

  1. neale revised this gist May 29, 2018. 1 changed file with 1 addition and 1 deletion.
    2 changes: 1 addition & 1 deletion plot_trajectory.py
    View file
    Original file line number Diff line number Diff line change
    @@ -38,6 +38,6 @@ def update_lines(num, data, line):
    ax.set_title('3D Test')

    # Creating the Animation object
    line_ani = animation.FuncAnimation(fig, update_lines, data.shape[1], fargs=(data, line), interval=50, blit=False)
    line_ani = animation.FuncAnimation(fig, update_lines, data.shape[1], fargs=(data, line), interval=50, blit=False)

    plt.show()
  2. neale created this gist May 29, 2018.
    43 changes: 43 additions & 0 deletions plot_trajectory.py
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,43 @@
    import numpy as np
    import matplotlib.pyplot as plt
    import mpl_toolkits.mplot3d.axes3d as p3
    import matplotlib.animation as animation
    import pandas as pd
    from sys import exit
    def update_lines(num, data, line):
    # NOTE: there is no .set_data() for 3 dim data...
    line.set_data(data[0:2, :num])
    line.set_3d_properties(data[2, :num])
    return line

    # Attaching 3D axis to the figure
    fig = plt.figure()
    ax = p3.Axes3D(fig)

    # Reading the data from a CSV file using pandas
    repo = pd.read_csv('coords2.csv',sep=',',header=0)
    data = np.array((repo['x'].values, repo['y'].values, repo['z'].values))
    print (data.shape[1])
    #exit()

    # Creating fifty line objects.
    # NOTE: Can't pass empty arrays into 3d version of plot()
    limit = 100000.
    line = ax.plot(data[0, 0:1], data[1, 0:1], data[2, 0:1])[0]

    # Setting the axes properties
    ax.set_xlim3d([-limit, limit])
    ax.set_xlabel('X')

    ax.set_ylim3d([-limit, limit])
    ax.set_ylabel('Y')

    ax.set_zlim3d([-limit, limit])
    ax.set_zlabel('Z')

    ax.set_title('3D Test')

    # Creating the Animation object
    line_ani = animation.FuncAnimation(fig, update_lines, data.shape[1], fargs=(data, line), interval=50, blit=False)

    plt.show()