|
22 | 22 | import matplotlib.pyplot as plt |
23 | 23 | import matplotlib.colors as colors |
24 | 24 |
|
| 25 | + |
| 26 | +def rbf(x, y): |
| 27 | + return 1.0 / (1 + 5 * ((x ** 2) + (y ** 2))) |
| 28 | + |
25 | 29 | N = 200 |
26 | 30 | gain = 8 |
27 | 31 | X, Y = np.mgrid[-3:3:complex(0, N), -2:2:complex(0, N)] |
28 | | -rbf = lambda x, y: 1.0 / (1 + 5 * ((x ** 2) + (y ** 2))) |
29 | 32 | Z1 = rbf(X + 0.5, Y + 0.5) |
30 | 33 | Z2 = rbf(X - 0.5, Y - 0.5) |
31 | 34 | Z = gain * Z1 - Z2 |
32 | 35 |
|
33 | | -shadeopts = { 'cmap': 'PRGn', 'shading': 'gouraud' } |
| 36 | +shadeopts = {'cmap': 'PRGn', 'shading': 'gouraud'} |
34 | 37 | colormap = 'PRGn' |
35 | 38 | lnrwidth = 0.2 |
36 | 39 |
|
|
66 | 69 |
|
67 | 70 | pcm = ax[0].pcolormesh(X, Y, Z, |
68 | 71 | norm=colors.SymLogNorm(linthresh=lnrwidth, linscale=1, |
69 | | - vmin=-gain, vmax=gain, base=10), |
| 72 | + vmin=-2, vmax=gain, base=10), |
70 | 73 | **shadeopts) |
71 | 74 | fig.colorbar(pcm, ax=ax[0], extend='both') |
72 | 75 | ax[0].text(-2.5, 1.5, 'symlog') |
73 | 76 |
|
74 | 77 | pcm = ax[1].pcolormesh(X, Y, Z, |
75 | 78 | norm=colors.AsinhNorm(linear_width=lnrwidth, |
76 | | - vmin=-gain, vmax=gain), |
| 79 | + vmin=-2, vmax=gain), |
77 | 80 | **shadeopts) |
78 | 81 | fig.colorbar(pcm, ax=ax[1], extend='both') |
79 | 82 | ax[1].text(-2.5, 1.5, 'asinh') |
|
0 commit comments