code: def save_plot(code, elem):
      
      if 'plt' in elem.attributes:
          figurewidth, figureheight = elem.attributes['plt'].split(',')
      else:
          try:
              figureheight = elem.attributes['height']
          except KeyError:
              figureheight = '4cm'
  
          try:
              figurewidth = elem.attributes['width']
          except KeyError:
              figurewidth = '6cm'
  
      return f"""import matplotlib
  matplotlib.use('TkAgg')
  {code}
  from matplotlib2tikz import get_tikz_code
  tikz = get_tikz_code(figureheight='{figureheight}', figurewidth='{figurewidth}')  # noqa
  print(tikz)"""
docs: Converts matplotlib plots to tikz code.
  
      If elem has either the plt attribute (format: plt=width,height) or the
      attributes width=width and/or height=height, the figurewidth and -height
      are set accordingly. If none are given, a height of 4cm and a width of 6cm
      is used as default.
  
      Args:
          code: The matplotlib code.
          elem: The element.
  
      Returns:
          The code and some code to invoke matplotlib2tikz.