演示轴 RGB

RGBAxes显示RGB合成图像。

演示轴 RGB

演示轴 RGB2

  1. import numpy as np
  2. import matplotlib.pyplot as plt
  4. from mpl_toolkits.axes_grid1.axes_rgb import make_rgb_axes, RGBAxes
  7. def get_demo_image():
  8.     from matplotlib.cbook import get_sample_data
  9.     f = get_sample_data("axes_grid/bivariate_normal.npy", asfileobj=False)
  10.     z = np.load(f)
  11.     # z is a numpy array of 15x15
  12.     return z, (-3, 4, -4, 3)
  15. def get_rgb():
  16.     Z, extent = get_demo_image()
  18.     Z[Z < 0] = 0.
  19.     Z = Z/Z.max()
  21.     R = Z[:13, :13]
  22.     G = Z[2:, 2:]
  23.     B = Z[:13, 2:]
  25.     return R, G, B
  28. def make_cube(r, g, b):
  29.     ny, nx = r.shape
  30.     R = np.zeros([ny, nx, 3], dtype="d")
  31.     R[:, :, 0] = r
  32.     G = np.zeros_like(R)
  33.     G[:, :, 1] = g
  34.     B = np.zeros_like(R)
  35.     B[:, :, 2] = b
  37.     RGB = R + G + B
  39.     return R, G, B, RGB
  42. def demo_rgb():
  43.     fig, ax = plt.subplots()
  44.     ax_r, ax_g, ax_b = make_rgb_axes(ax, pad=0.02)
  45.     #fig.add_axes(ax_r)
  46.     #fig.add_axes(ax_g)
  47.     #fig.add_axes(ax_b)
  49.     r, g, b = get_rgb()
  50.     im_r, im_g, im_b, im_rgb = make_cube(r, g, b)
  51.     kwargs = dict(origin="lower", interpolation="nearest")
  52.     ax.imshow(im_rgb, **kwargs)
  53.     ax_r.imshow(im_r, **kwargs)
  54.     ax_g.imshow(im_g, **kwargs)
  55.     ax_b.imshow(im_b, **kwargs)
  58. def demo_rgb2():
  59.     fig = plt.figure(2)
  60.     ax = RGBAxes(fig, [0.1, 0.1, 0.8, 0.8], pad=0.0)
  61.     #fig.add_axes(ax)
  62.     #ax.add_RGB_to_figure()
  64.     r, g, b = get_rgb()
  65.     kwargs = dict(origin="lower", interpolation="nearest")
  66.     ax.imshow_rgb(r, g, b, **kwargs)
  68.     ax.RGB.set_xlim(0., 9.5)
  69.     ax.RGB.set_ylim(0.9, 10.6)
  71.     for ax1 in [ax.RGB, ax.R, ax.G, ax.B]:
  72.         for sp1 in ax1.spines.values():
  73.             sp1.set_color("w")
  74.         for tick in ax1.xaxis.get_major_ticks() + ax1.yaxis.get_major_ticks():
  75.             tick.tick1line.set_mec("w")
  76.             tick.tick2line.set_mec("w")
  78.     return ax
  81. demo_rgb()
  82. ax = demo_rgb2()
  84. plt.show()

下载这个示例