演示轴 Hbox Divider

HBox Divider用于排列子图。

演示轴 Hbox Divider

  1. import numpy as np
  2. import matplotlib.pyplot as plt
  3. from mpl_toolkits.axes_grid1.axes_divider import HBoxDivider
  4. import mpl_toolkits.axes_grid1.axes_size as Size
  7. def make_heights_equal(fig, rect, ax1, ax2, pad):
  8.     # pad in inches
  10.     h1, v1 = Size.AxesX(ax1), Size.AxesY(ax1)
  11.     h2, v2 = Size.AxesX(ax2), Size.AxesY(ax2)
  13.     pad_v = Size.Scaled(1)
  14.     pad_h = Size.Fixed(pad)
  16.     my_divider = HBoxDivider(fig, rect,
  17.                              horizontal=[h1, pad_h, h2],
  18.                              vertical=[v1, pad_v, v2])
  20.     ax1.set_axes_locator(my_divider.new_locator(0))
  21.     ax2.set_axes_locator(my_divider.new_locator(2))
  24. if __name__ == "__main__":
  26.     arr1 = np.arange(20).reshape((4, 5))
  27.     arr2 = np.arange(20).reshape((5, 4))
  29.     fig, (ax1, ax2) = plt.subplots(1, 2)
  30.     ax1.imshow(arr1, interpolation="nearest")
  31.     ax2.imshow(arr2, interpolation="nearest")
  33.     rect = 111  # subplot param for combined axes
  34.     make_heights_equal(fig, rect, ax1, ax2, pad=0.5)  # pad in inches
  36.     for ax in [ax1, ax2]:
  37.         ax.locator_params(nbins=4)
  39.     # annotate
  40.     ax3 = plt.axes([0.5, 0.5, 0.001, 0.001], frameon=False)
  41.     ax3.xaxis.set_visible(False)
  42.     ax3.yaxis.set_visible(False)
  43.     ax3.annotate("Location of two axes are adjusted\n"
  44.                  "so that they have equal heights\n"
  45.                  "while maintaining their aspect ratios", (0.5, 0.5),
  46.                  xycoords="axes fraction", va="center", ha="center",
  47.                  bbox=dict(boxstyle="round, pad=1", fc="w"))
  49.     plt.show()

下载这个示例