简单的轴垫

简单的轴垫

  1. import numpy as np
  2. import matplotlib.pyplot as plt
  3. import mpl_toolkits.axisartist.angle_helper as angle_helper
  4. import mpl_toolkits.axisartist.grid_finder as grid_finder
  5. from matplotlib.projections import PolarAxes
  6. from matplotlib.transforms import Affine2D
  8. import mpl_toolkits.axisartist as axisartist
  10. from mpl_toolkits.axisartist.grid_helper_curvelinear import \
  11.     GridHelperCurveLinear
  14. def setup_axes(fig, rect):
  15.     """
  16.     polar projection, but in a rectangular box.
  17.     """
  19.     # see demo_curvelinear_grid.py for details
  20.     tr = Affine2D().scale(np.pi/180., 1.) + PolarAxes.PolarTransform()
  22.     extreme_finder = angle_helper.ExtremeFinderCycle(20, 20,
  23.                                                      lon_cycle=360,
  24.                                                      lat_cycle=None,
  25.                                                      lon_minmax=None,
  26.                                                      lat_minmax=(0, np.inf),
  27.                                                      )
  29.     grid_locator1 = angle_helper.LocatorDMS(12)
  30.     grid_locator2 = grid_finder.MaxNLocator(5)
  32.     tick_formatter1 = angle_helper.FormatterDMS()
  34.     grid_helper = GridHelperCurveLinear(tr,
  35.                                         extreme_finder=extreme_finder,
  36.                                         grid_locator1=grid_locator1,
  37.                                         grid_locator2=grid_locator2,
  38.                                         tick_formatter1=tick_formatter1
  39.                                         )
  41.     ax1 = axisartist.Subplot(fig, rect, grid_helper=grid_helper)
  42.     ax1.axis[:].set_visible(False)
  44.     fig.add_subplot(ax1)
  46.     ax1.set_aspect(1.)
  47.     ax1.set_xlim(-5, 12)
  48.     ax1.set_ylim(-5, 10)
  50.     return ax1
  53. def add_floating_axis1(ax1):
  54.     ax1.axis["lat"] = axis = ax1.new_floating_axis(0, 30)
  55.     axis.label.set_text(r"$\theta = 30^{\circ}$")
  56.     axis.label.set_visible(True)
  58.     return axis
  61. def add_floating_axis2(ax1):
  62.     ax1.axis["lon"] = axis = ax1.new_floating_axis(1, 6)
  63.     axis.label.set_text(r"$r = 6$")
  64.     axis.label.set_visible(True)
  66.     return axis
  69. fig = plt.figure(1, figsize=(9, 3.))
  70. fig.clf()
  71. fig.subplots_adjust(left=0.01, right=0.99, bottom=0.01, top=0.99,
  72.                     wspace=0.01, hspace=0.01)
  75. def ann(ax1, d):
  76.     if plt.rcParams["text.usetex"]:
  77.         d = d.replace("_", r"\_")
  79.     ax1.annotate(d, (0.5, 1), (5, -5),
  80.                  xycoords="axes fraction", textcoords="offset points",
  81.                  va="top", ha="center")
  84. ax1 = setup_axes(fig, rect=141)
  85. axis = add_floating_axis1(ax1)
  86. ann(ax1, r"default")
  88. ax1 = setup_axes(fig, rect=142)
  89. axis = add_floating_axis1(ax1)
  90. axis.major_ticklabels.set_pad(10)
  91. ann(ax1, r"ticklabels.set_pad(10)")
  93. ax1 = setup_axes(fig, rect=143)
  94. axis = add_floating_axis1(ax1)
  95. axis.label.set_pad(20)
  96. ann(ax1, r"label.set_pad(20)")
  98. ax1 = setup_axes(fig, rect=144)
  99. axis = add_floating_axis1(ax1)
  100. axis.major_ticks.set_tick_out(True)
  101. ann(ax1, "ticks.set_tick_out(True)")
  103. plt.show()

下载这个示例