Python 实现火柴人的设计与实现

bash复制代码
pip install matplotlib

import matplotlib.pyplot as plt  import numpy as np

# 定义关节位置  head = [0, 1]  torso = [0, 0]  left_shoulder = [-0.5, 0]  left_elbow = [-1, -0.5]  left_hand = [-1, -1]  right_shoulder = [0.5, 0]  right_elbow = [1, -0.5]  right_hand = [1, -1]  left_hip = [-0.5, -0.5]  left_knee = [-1, -1.5]  left_foot = [-1, -2]  right_hip = [0.5, -0.5]  right_knee = [1, -1.5]  right_foot = [1, -2]    # 将关节位置存储在一个字典中  joints = {      'head': head,      'torso': torso,      'left_shoulder': left_shoulder,      'left_elbow': left_elbow,      'left_hand': left_hand,      'right_shoulder': right_shoulder,      'right_elbow': right_elbow,      'right_hand': right_hand,      'left_hip': left_hip,      'left_knee': left_knee,      'left_foot': left_foot,      'right_hip': right_hip,      'right_knee': right_knee,      'right_foot': right_foot  }

def draw_stick_figure(joints, ax):      # 绘制身体      body_parts = [          ('torso', 'head'),          ('torso', 'left_shoulder'), ('left_shoulder', 'left_elbow'), ('left_elbow', 'left_hand'),          ('torso', 'right_shoulder'), ('right_shoulder', 'right_elbow'), ('right_elbow', 'right_hand'),          ('torso', 'left_hip'), ('left_hip', 'left_knee'), ('left_knee', 'left_foot'),          ('torso', 'right_hip'), ('right_hip', 'right_knee'), ('right_knee', 'right_foot')      ]            for start, end in body_parts:          start_pos = np.array(joints[start])          end_pos = np.array(joints[end])          ax.plot([start_pos[0], end_pos[0]], [start_pos[1], end_pos[1]], 'k-')            # 绘制头部      circle = plt.Circle(joints['head'], 0.1, color='black', fill=True)      ax.add_patch(circle)            # 绘制手部（可选）      circle = plt.Circle(joints['left_hand'], 0.05, color='black', fill=True)      ax.add_patch(circle)      circle = plt.Circle(joints['right_hand'], 0.05, color='black', fill=True)      ax.add_patch(circle)            # 绘制脚部（可选）      circle = plt.Circle(joints['left_foot'], 0.05, color='black', fill=True)      ax.add_patch(circle)      circle = plt.Circle(joints['right_foot'], 0.05, color='black', fill=True)      ax.add_patch(circle)

def main():      fig, ax = plt.subplots()      ax.set_aspect('equal')      ax.axis('off')  # 关闭坐标轴            draw_stick_figure(joints, ax)            plt.show()    if __name__ == "__main__":      main()

import matplotlib.animation as animation    def update_position(frame, joints):      # 这里我们简单地将火柴人向右移动      translation = 0.1 * frame      for key in joints.keys():          joints[key][0] += translation      return joints    def animate(frame):      global joints_anim      joints_anim = update_position(frame, joints_anim)      ax.clear()      ax.set_aspect('equal')      ax.axis('off')      draw_stick_figure(joints_anim, ax)    def main_animation():      fig, ax = plt.subplots()      global joints_anim      joints_anim = {key: value.copy() for key, value in joints.items()}  # 复制初始关节位置      ani = animation.FuncAnimation(fig, animate, frames=100, interval=100)      plt.show()    if __name__ == "__main__":      main_animation()

import matplotlib.pyplot as plt  import numpy as np  import matplotlib.animation as animation    # 定义关节位置  joints = {      'head': [0, 1],      'torso': [0, 0],      'left_shoulder': [-0.5, 0],      'left_elbow': [-1, -0.5],      'left_hand': [-1, -1],      'right_shoulder': [0.5, 0],      'right_elbow': [1, -0.5],      'right_hand': [1, -1],      'left_hip': [-0.5, -0.5],      'left_knee': [-1, -1.5],      'left_foot': [-1, -2],      'right_hip': [0.5, -0.5],      'right_knee': [1, -1.5],      'right_foot': [1, -2]  }    # 将关节位置转换为numpy数组，以便进行数学运算  joints = {key: np.array(value) for key, value in joints.items()}    # 绘制火柴人的函数  def draw_stick_figure(joints, ax):      # 清除之前的绘图      ax.clear()            # 设置坐标轴的比例和限制      ax.set_aspect('equal')      ax.set_xlim(-2, 2)      ax.set_ylim(-2.5, 1.5)            # 定义身体部分和对应的颜色（可选）      body_parts = [          ('torso', 'head', 'black'),          ('torso', 'left_shoulder', 'black'), ('left_shoulder', 'left_elbow', 'black'), ('left_elbow', 'left_hand', 'black'),          ('torso', 'right_shoulder', 'black'), ('right_shoulder', 'right_elbow', 'black'), ('right_elbow', 'right_hand', 'black'),          ('torso', 'left_hip', 'black'), ('left_hip', 'left_knee', 'black'), ('left_knee', 'left_foot', 'black'),          ('torso', 'right_hip', 'black'), ('right_hip', 'right_knee', 'black'), ('right_knee', 'right_foot', 'black')      ]            # 绘制火柴人的各个部分      for part in body_parts:          start_joint, end_joint, color = part[0], part[1], part[2] if len(part) > 2 else 'black'          ax.plot([joints[start_joint][0], joints[end_joint][0]], [joints[start_joint][1], joints[end_joint][1]], color=color, linewidth=2)            # 显示网格（可选）      ax.grid(True)    # 创建图形和坐标轴  fig, ax = plt.subplots()    # 初始化函数（用于动画）  def init():      draw_stick_figure(joints, ax)      return []  # 返回空列表，因为我们没有需要更新的艺术家对象    # 动画更新函数  def update(frame):      # 这里可以添加使火柴人移动或改变姿势的逻辑      # 例如，简单地旋转手臂或腿      # 但为了简化，我们在这里不改变关节位置      draw_stick_figure(joints, ax)      return []  # 同样返回空列表    # 创建动画  ani = animation.FuncAnimation(fig, update, frames=100, init_func=init, blit=True, interval=100)    # 显示图形  plt.show()