演示浮动轴

浮动轴的演示。

  1. from matplotlib.transforms import Affine2D
  2. import mpl_toolkits.axisartist.floating_axes as floating_axes
  3. import numpy as np
  4. import mpl_toolkits.axisartist.angle_helper as angle_helper
  5. from matplotlib.projections import PolarAxes
  6. from mpl_toolkits.axisartist.grid_finder import (FixedLocator, MaxNLocator,
  7.                                                  DictFormatter)
  8. import matplotlib.pyplot as plt
  10. # Fixing random state for reproducibility
  11. np.random.seed(19680801)
  14. def setup_axes1(fig, rect):
  15.     """
  16.     A simple one.
  17.     """
  18.     tr = Affine2D().scale(2, 1).rotate_deg(30)
  20.     grid_helper = floating_axes.GridHelperCurveLinear(
  21.         tr, extremes=(-0.5, 3.5, 0, 4))
  23.     ax1 = floating_axes.FloatingSubplot(fig, rect, grid_helper=grid_helper)
  24.     fig.add_subplot(ax1)
  26.     aux_ax = ax1.get_aux_axes(tr)
  28.     grid_helper.grid_finder.grid_locator1._nbins = 4
  29.     grid_helper.grid_finder.grid_locator2._nbins = 4
  31.     return ax1, aux_ax
  34. def setup_axes2(fig, rect):
  35.     """
  36.     With custom locator and formatter.
  37.     Note that the extreme values are swapped.
  38.     """
  39.     tr = PolarAxes.PolarTransform()
  41.     pi = np.pi
  42.     angle_ticks = [(0, r"$0$"),
  43.                    (.25*pi, r"$\frac{1}{4}\pi$"),
  44.                    (.5*pi, r"$\frac{1}{2}\pi$")]
  45.     grid_locator1 = FixedLocator([v for v, s in angle_ticks])
  46.     tick_formatter1 = DictFormatter(dict(angle_ticks))
  48.     grid_locator2 = MaxNLocator(2)
  50.     grid_helper = floating_axes.GridHelperCurveLinear(
  51.         tr, extremes=(.5*pi, 0, 2, 1),
  52.         grid_locator1=grid_locator1,
  53.         grid_locator2=grid_locator2,
  54.         tick_formatter1=tick_formatter1,
  55.         tick_formatter2=None)
  57.     ax1 = floating_axes.FloatingSubplot(fig, rect, grid_helper=grid_helper)
  58.     fig.add_subplot(ax1)
  60.     # create a parasite axes whose transData in RA, cz
  61.     aux_ax = ax1.get_aux_axes(tr)
  63.     aux_ax.patch = ax1.patch  # for aux_ax to have a clip path as in ax
  64.     ax1.patch.zorder = 0.9  # but this has a side effect that the patch is
  65.     # drawn twice, and possibly over some other
  66.     # artists. So, we decrease the zorder a bit to
  67.     # prevent this.
  69.     return ax1, aux_ax
  72. def setup_axes3(fig, rect):
  73.     """
  74.     Sometimes, things like axis_direction need to be adjusted.
  75.     """
  77.     # rotate a bit for better orientation
  78.     tr_rotate = Affine2D().translate(-95, 0)
  80.     # scale degree to radians
  81.     tr_scale = Affine2D().scale(np.pi/180., 1.)
  83.     tr = tr_rotate + tr_scale + PolarAxes.PolarTransform()
  85.     grid_locator1 = angle_helper.LocatorHMS(4)
  86.     tick_formatter1 = angle_helper.FormatterHMS()
  88.     grid_locator2 = MaxNLocator(3)
  90.     # Specify theta limits in degrees
  91.     ra0, ra1 = 8.*15, 14.*15
  92.     # Specify radial limits
  93.     cz0, cz1 = 0, 14000
  94.     grid_helper = floating_axes.GridHelperCurveLinear(
  95.         tr, extremes=(ra0, ra1, cz0, cz1),
  96.         grid_locator1=grid_locator1,
  97.         grid_locator2=grid_locator2,
  98.         tick_formatter1=tick_formatter1,
  99.         tick_formatter2=None)
  101.     ax1 = floating_axes.FloatingSubplot(fig, rect, grid_helper=grid_helper)
  102.     fig.add_subplot(ax1)
  104.     # adjust axis
  105.     ax1.axis["left"].set_axis_direction("bottom")
  106.     ax1.axis["right"].set_axis_direction("top")
  108.     ax1.axis["bottom"].set_visible(False)
  109.     ax1.axis["top"].set_axis_direction("bottom")
  110.     ax1.axis["top"].toggle(ticklabels=True, label=True)
  111.     ax1.axis["top"].major_ticklabels.set_axis_direction("top")
  112.     ax1.axis["top"].label.set_axis_direction("top")
  114.     ax1.axis["left"].label.set_text(r"cz [km$^{-1}$]")
  115.     ax1.axis["top"].label.set_text(r"$\alpha_{1950}$")
  117.     # create a parasite axes whose transData in RA, cz
  118.     aux_ax = ax1.get_aux_axes(tr)
  120.     aux_ax.patch = ax1.patch  # for aux_ax to have a clip path as in ax
  121.     ax1.patch.zorder = 0.9  # but this has a side effect that the patch is
  122.     # drawn twice, and possibly over some other
  123.     # artists. So, we decrease the zorder a bit to
  124.     # prevent this.
  126.     return ax1, aux_ax
  1. fig = plt.figure(1, figsize=(8, 4))
  2. fig.subplots_adjust(wspace=0.3, left=0.05, right=0.95)
  4. ax1, aux_ax1 = setup_axes1(fig, 131)
  5. aux_ax1.bar([0, 1, 2, 3], [3, 2, 1, 3])
  7. ax2, aux_ax2 = setup_axes2(fig, 132)
  8. theta = np.random.rand(10)*.5*np.pi
  9. radius = np.random.rand(10) + 1.
  10. aux_ax2.scatter(theta, radius)
  12. ax3, aux_ax3 = setup_axes3(fig, 133)
  14. theta = (8 + np.random.rand(10)*(14 - 8))*15.  # in degrees
  15. radius = np.random.rand(10)*14000.
  16. aux_ax3.scatter(theta, radius)
  18. plt.show()

演示浮动轴

下载这个示例