使用matplotlib绘制动图

主程序如下:

python 复制代码
'''
项目:701项目三分报告
作者:WN
内容:速度障碍法实现
时间:2023年8月26号
'''
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
from matplotlib import animation
from utils import *

uav1 = Agent(1, [0, 0])
trace1 = []
uav2 = Agent(2, [40, 40])
uav2.yaw = uav1.yaw + np.pi
trace2 = []

trace = []


for i in range(100):
    uav1.step()
    trace1.append((uav1.pos[0], uav1.pos[1]))
    uav2.step()
    trace2.append((uav2.pos[0], uav2.pos[1]))
    trace.append((uav1.pos[0], uav1.pos[1], uav2.pos[0], uav2.pos[1]))

# plt.xlim(0,40)
# plt.ylim(0,40)
# plt.scatter([x[0] for x in trace1], [x[1] for x in trace1], color='r')
# plt.scatter([x[0] for x in trace2], [x[1] for x in trace2], color='b')
# plt.show()


fig, ax = plt.subplots()
xdata1, ydata1 = [], []
xdata2, ydata2 = [], []

line1, = ax.plot(xdata1, ydata1, 'ro')
line2, = ax.plot(xdata2, ydata2, 'bo')

def init():
    ax.set_xlim(0,40)
    ax.set_ylim(0,40)
    return line1, line2
def update(frame):  # 帧
    print(frame)
    xdata1.append(frame[0])
    ydata1.append(frame[1])
    line1.set_data(xdata1, ydata1)

    xdata2.append(frame[2])
    ydata2.append(frame[3])
    line2.set_data(xdata2, ydata2)
    return line1,line2

ani = animation.FuncAnimation(
    fig=fig,
    func=update,
    frames=trace,
    init_func=init,
    interval=100, # 时间间隔
)
plt.show()
ani.save('1.gif', writer='pillow')

函数库如下:

python 复制代码
'''
作者:WN
内容:速度障碍法实现的主要函数
时间:2023年8月26号
'''
import numpy as np


class Agent(object):
    def __init__(self, ID, pos_0):
        self.id = ID
        self.v = 0.05  # 平台的速度(米/秒)
        self.pos = pos_0  # 平台的初始位置
        self.yaw = np.pi/3  # 平台的航向(0至2pi)
        self.det = 5  # 平台的探测范围
        self.state = 'None'  # 平台的状态(暂为缺省值)
        self.step_num = 0  # 记录平台仿真时间
        self.obs = None  # 记录平台所能观测到的数据

    def Stragey(self):
        '''
        根据智能体得到的观测数据来决策(输入、输出应根据需求具体设计)
        输入: 观测数据、当前位置、探测数据
        输出:新的速度
        '''
        pass

    def Pos_update(self):
        '''
        输入:当前位置、速度
        输出:更新后的位置
        '''
        self.pos[0] += self.v * np.cos(self.yaw)
        self.pos[1] += self.v * np.sin(self.yaw)
        pass

    def step(self):
        self.step_num += 1

        #########根据探测到的数据和策略解算新的位置、速度
        # self.v = self.Stragey()
        self.Pos_update()
相关推荐
线条12 天前
Matplotlib 安装部署与版本兼容问题解决方案(pyCharm)
matplotlib
HuashuiMu花水木2 天前
Matplotlib笔记4----------图像处理
图像处理·笔记·matplotlib
程序员阿超的博客4 天前
Python 数据分析与机器学习入门 (五):Matplotlib 数据可视化基础
python·信息可视化·数据分析·matplotlib·数据可视化·python教程·pyplot
音程15 天前
Matplotlib绘制矩阵图,plt.matshow/imshow 与 ax.pcolor(pcolormesh)方法的使用
matplotlib
搏博16 天前
基于Python、tkinter、sqlite3 和matplotlib的校园书店管理系统
python·信息可视化·sqlite·matplotlib
勤奋的知更鸟18 天前
Matplotlib 绘图库使用技巧介绍
开发语言·python·matplotlib
@小红花19 天前
Matplotlib快速入门
matplotlib
仟濹22 天前
「Matplotlib 入门指南」 Python 数据可视化分析【数据分析全栈攻略:爬虫+处理+可视化+报告】
python·信息可视化·数据分析·matplotlib
Jay_271 个月前
cloudstudio腾讯云:matplotlib 设置中文字体
腾讯云·matplotlib
Gyoku Mint1 个月前
机器学习×第二卷:概念下篇——她不再只是模仿,而是开始决定怎么靠近你
人工智能·python·算法·机器学习·pandas·ai编程·matplotlib