Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- fig = plt.figure(figsize = (13,5))
- axes = fig.subplot_mosaic(
- """
- ABCDEF
- GGHHII
- GGHHII
- """
- )
- axes = [axes[k] for k in axes.keys()]
- n = 1024
- arr = np.zeros((n,n))
- l = np.linspace(-1, 1, n)
- nr = np.sqrt(l[:, None]**2 + l[None, :]**2)
- thicknesses = []
- for r in np.linspace(0.3, 0.9, 6):
- theta = np.arcsin(nr/r)
- thickness = 0.3/(np.cos(theta)+0.3)
- thickness = np.nan_to_num(thickness, 0)
- thicknesses.append(thickness)
- for i, (t, ax) in enumerate(zip(thicknesses, axes)):
- ax.imshow(t)
- ax.set_title(f'img {i}')
- cim = axes[6].imshow(np.sum(thicknesses, axis = 0))
- cbar = fig.colorbar(cim)
- cbar.set_ticks([])
- axes[6].set_title('sum')
- cmap = mpl.multivar_colormaps['6VarAddA']
- cim = axes[7].imshow(thicknesses, cmap = cmap)
- axes[7].set_title("combination_mode = 'sRGB_add'")
- cbars = fig.colorbars(cim, (-1, 2), fraction_per_row=0.08)
- for cbar in cbars:
- cbar.set_ticks([])
- ims = np.array([c(t) for c, t in zip(cmap, thicknesses)])
- im = np.choose(np.argmax(thicknesses, axis = 0).reshape(-1,1), ims.reshape((len(cmap),-1,4))).reshape((*thicknesses[0].shape, -1))
- cim = axes[8].imshow(im)
- axes[8].set_title("combination_mode = 'choose_max'")
- for ax in axes:
- ax.set_yticks([])
- ax.set_xticks([])
Add Comment
Please, Sign In to add comment
