()
| 786 | from mpl_toolkits.axes_grid1.axes_rgb import RGBAxes |
| 787 | |
| 788 | def get_demo_image(): |
| 789 | # prepare image |
| 790 | delta = 0.5 |
| 791 | |
| 792 | extent = (-3, 4, -4, 3) |
| 793 | x = np.arange(-3.0, 4.001, delta) |
| 794 | y = np.arange(-4.0, 3.001, delta) |
| 795 | X, Y = np.meshgrid(x, y) |
| 796 | Z1 = np.exp(-X ** 2 - Y ** 2) |
| 797 | Z2 = np.exp(-(X - 1) ** 2 - (Y - 1) ** 2) |
| 798 | Z = (Z1 - Z2) * 2 |
| 799 | |
| 800 | return Z, extent |
| 801 | |
| 802 | def get_rgb(): |
| 803 | Z, extent = get_demo_image() |