3D轨迹图的绘制及其gif保存

绘制三维轨迹图

单纯的画3D轨迹图相对来说比较简单，也有很多种方法，这里主要用Axes3D去画

1. 首先导入要用的模块

from typing import List
import matplotlib.pyplot as plt
import mpl_toolkits.mplot3d.axes3d as p3

2. 建立画布

fig = plt.figure()
ax = p3.Axes3D(fig)

3. 导入数据，这里可以根据自己的需要去改，我这里是自己随便写了几个list，也可以自己从csv或者其他地方读取数据，改一下就可以了

high: List[float] = [800,820,801,801,803,815]
longitude: List[float] = [100,103,105,107,109,112]
latitude: List[float] = [60,62,65,66,67,69]
z: List[float] = [800,801,801,811,803,815]
x: List[float] = [200,189,178,176,175,172]
y: List[float] = [130,129,128,127,126,124]

4. 轨迹线绘制

ax.plot(latitude, longitude, high, color='red')
line_1 = ax.plot(latitude, longitude, high, color='red')
ax.plot(x, y, z, color='blue')
line_2 = ax.plot(x, y, z, color='blue')
lines = [line_1[0], line_2[0]]

5. 坐标轴设置及最终效果显示

ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('Z')
ax.set_title('Trajectory of the 3D')
plt.show()

最后放上全部的代码大家复制粘贴安装包就可以直接跑了

# -*- coding: utf-8 -*-
"""
Created on Sun Apr 26 12:49:00 2020

@author: yunhui
"""
from typing import List
import matplotlib.pyplot as plt
import mpl_toolkits.mplot3d.axes3d as p3




fig = plt.figure()
ax = p3.Axes3D(fig)
 # 绘制出两条轨迹线
high: List[float] = [800,820,801,801,803,815]
longitude: List[float] = [100,103,105,107,109,112]
latitude: List[float] = [60,62,65,66,67,69]

z: List[float] = [800,801,801,811,803,815]
x: List[float] = [200,189,178,176,175,172]
y: List[float] = [130,129,128,127,126,124]

ax.plot(latitude, longitude, high, color='red')
line_1 = ax.plot(latitude, longitude, high, color='red')
ax.plot(x, y, z, color='blue')
line_2 = ax.plot(x, y, z, color='blue')
lines = [line_1[0], line_2[0]]

ax.set_xlabel('X')

ax.set_ylabel('Y')

ax.set_zlabel('Z')

ax.set_title('Trajectory of the 3D')
plt.show()

3D轨迹图绘制gif图片保存

主要就是多了一个gif的渲染，可以通过调节渲染的帧率等来控制轨迹移动的快慢等，直接给大家放代码和效果图

# -*- coding: utf-8 -*-
"""
Created on Sun May 10 18:10:41 2020

@author: yunhui
"""

from typing import List
import matplotlib.pyplot as plt
import mpl_toolkits.mplot3d.axes3d as p3
import matplotlib.animation as animation


def update_lines(num, dataLines, lines):
    for line, data in zip(lines, dataLines):
        line.set_data(data[0][num], data[1][num])
        line.set_3d_properties(data[2][num])
    return lines

fig = plt.figure()
ax = p3.Axes3D(fig)
 # 绘制出两条轨迹线
high: List[float] =      [8000,8010,8015,8020,8010,8011,8013,8015,8017,8000]
longitude: List[float] = [100, 101, 102, 103, 105, 107, 109, 112, 115, 119]
latitude: List[float] =  [60,  61,  62,  63,  65,  66,  67,  69,  70,  72]
ax.plot(latitude, longitude, high, color='red')
line_1 = ax.plot(latitude, longitude, high,marker='o', color='red')
z: List[float] =         [8000,8005,8007,8010,8010,8011,8003,8015,8007,8000]
y: List[float] =         [130, 129, 129, 129, 128, 127, 126, 124, 123, 121]
x: List[float] =         [200, 195, 192, 189, 178, 176, 175, 172, 173, 172]
ax.plot(x, y, z, color='blue') 
line_2 = ax.plot(x, y, z,marker='o', color='blue')
lines = [line_1[0], line_2[0]]

ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('Z')
ax.set_title('Trajectory of the planes')

length_data = len(latitude)
 # 选择 5 个关键节点
select_index = range(0, length_data, length_data // 5)
select_la: List[float] = [latitude[i] for i in select_index]
select_lo: List[float] = [longitude[i] for i in select_index]
select_al: List[float] = [high[i] for i in select_index]
select_x: List[float] = [x[i] for i in select_index]
select_y : List[float] = [y[i] for i in select_index]
select_z: List[float] = [z[i] for i in select_index]
total_data = [[select_la, select_lo, select_al], [select_x, select_y, select_z]]
len_select: int = len(select_x)
# Creating the Animation object
line_ani = animation.FuncAnimation(fig, update_lines, len_select, fargs=(total_data, lines),
                                           interval = 1, blit=False)
#把一些特俗的点标出来
ax.scatter(latitude[1], longitude[1], high[1], s=100,c='r',marker='*')

plt.show()
line_ani.save('plot.gif', writer='pillow',fps=2)