Python中是否有类似其他语言的with语句以简化matplotlib调用？

Python里确实没有VBA/.NET那种直接省略对象名的With语法，但有几种实用方法可以减少重复的plt调用：

• 简化导入名称
  把matplotlib.pyplot导入成更短的别名，是最简单直接的方式：

  import matplotlib.pyplot as p
  
  p.xlabel("Time")
  p.ylabel("Amplitude")
  

• 利用Axes对象的链式调用
  matplotlib的Axes类方法大多会返回自身实例，所以可以链式调用设置属性：

  import matplotlib.pyplot as plt
  
  # 获取当前Axes对象后链式调用
  ax = plt.gca()
  ax.set_xlabel("Time").set_ylabel("Amplitude")
  
  # 或者创建子图时直接操作
  ax = plt.subplots()[1]
  ax.set_xlabel("Time").set_ylabel("Amplitude")
  

  注意：plt.xlabel()这类全局函数本质也是调用当前Axes的方法，直接操作Axes对象是更符合matplotlib面向对象设计的方式。

• 自定义上下文管理器模拟With效果
  如果你非要接近VBA的With写法，可以自己写个极简的上下文管理器：

  import matplotlib.pyplot as plt
  
  class WithWrapper:
      def __init__(self, obj):
          self.obj = obj
      def __enter__(self):
          return self.obj
      def __exit__(self, *args):
          pass
  
  # 使用时和你期望的形式接近，只是需要显式写出变量名
  with WithWrapper(plt) as p:
      p.xlabel("Time")
      p.ylabel("Amplitude")
  

  不过这种写法和用别名的区别不大，只是多了上下文的形式，Python原生不支持省略对象名的链式写法。

• 封装成复用函数
  如果有大量重复的图表配置操作，直接封装成函数更高效：

  import matplotlib.pyplot as plt
  
  def setup_plot_labels(x_text, y_text):
      plt.xlabel(x_text)
      plt.ylabel(y_text)
  
  # 调用时一行搞定
  setup_plot_labels("Time", "Amplitude")
  

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