Mac OS X+Anaconda2.7下Matplotlib FuncAnimation两大问题求助

我之前在Mac OS X上用Anaconda Python2.7跑Matplotlib的FuncAnimation时也踩过一模一样的坑！给你分享两个问题的具体解决方案：

一、让FuncAnimation优雅结束 #

动画结束后程序挂起甚至^C都无效，主要是因为Matplotlib的MacOSX后端在事件循环处理上的特性，加上动画资源没正确释放导致的。可以试试这些方法：

• 保存动画对象引用并主动停止+关闭窗口
  一定要把FuncAnimation的实例保存下来（比如赋值给ani），不然Python的垃圾回收会提前回收它，引发异常。然后在结束条件触发时，先停止动画事件源，再关闭绘图窗口：

  import matplotlib.pyplot as plt
  from matplotlib.animation import FuncAnimation
  
  fig, ax = plt.subplots()
  line, = ax.plot([0, 1], [0, 1])
  
  def update(frame):
      line.set_ydata([frame/10.0, frame/10.0])
      # 自定义结束条件，比如帧到达20时停止
      if frame >= 20:
          ani.event_source.stop()  # 停止动画更新
          plt.close(fig)  # 关闭窗口释放资源
      return line,
  
  ani = FuncAnimation(fig, update, frames=range(30), interval=100)
  plt.show()
  

• 切换Matplotlib后端
  MacOSX默认的后端有时候会有阻塞问题，试试切换到更稳定的TkAgg后端（要在导入pyplot之前设置）：

  import matplotlib
  matplotlib.use('TkAgg')  # 必须放在import matplotlib.pyplot之前
  
  import matplotlib.pyplot as plt
  from matplotlib.animation import FuncAnimation
  # 后续代码不变
  

• 避免阻塞式显示的额外处理
  如果是在脚本中运行，plt.show()默认是阻塞的，动画结束后可以手动添加事件循环退出的逻辑，或者在动画停止后调用plt.close()来主动结束程序。

二、让key_press_event正常工作 #

键盘事件失效的核心原因大多是后端的焦点问题，或者事件绑定的时机不对，试试这些步骤：

• 正确绑定事件并确保动画对象可访问
  要在plt.show()之前绑定键盘事件，并且确保回调函数能访问到动画实例ani（比如用全局变量或者闭包）：

  import matplotlib
  matplotlib.use('TkAgg')  # 先切换后端，避免MacOSX后端的焦点问题
  
  import matplotlib.pyplot as plt
  from matplotlib.animation import FuncAnimation
  
  fig, ax = plt.subplots()
  line, = ax.plot([0, 1], [0, 1])
  
  def update(frame):
      line.set_ydata([frame/10.0, frame/10.0])
      return line,
  
  def on_key_press(event):
      print(f"Pressed key: {event.key}")
      if event.key == 'q':  # 按q键停止动画并退出
          ani.event_source.stop()
          plt.close(fig)
  
  ani = FuncAnimation(fig, update, frames=range(30), interval=100)
  fig.canvas.mpl_connect('key_press_event', on_key_press)  # 绑定事件
  plt.show()
  

• 解决窗口焦点问题
  MacOSX后端下，绘图窗口可能无法自动获得键盘焦点，导致事件无法被捕获。切换到TkAgg或者QtAgg后端（如果Anaconda中安装了PyQt）就能解决这个问题，上面的代码已经包含了切换到TkAgg的步骤。

• 检查官方示例的运行环境
  如果官方示例也挂起，基本可以确定是后端的问题，按照上面的方法切换后端后再运行示例，应该就能正常捕获键盘事件并优雅退出了。

内容的提问来源于stack exchange，提问作者Prof Huster