Extend plugins with imshow_collection

A first implementation is made for Matplotlib.
This commit is contained in:
Stefan van der Walt
2016-01-24 21:38:21 -08:00
parent 0d7289ea76
commit 3ffca2fdbd
7 changed files with 54 additions and 10 deletions
+13
View File
@@ -147,6 +147,7 @@ def imshow(im, ax=None, show_cbar=None, **kwargs):
if kwargs.get('cmap', None) == 'viridis':
kwargs['cmap'] = viridis
lo, hi, cmap = _get_display_range(im)
kwargs.setdefault('interpolation', 'nearest')
kwargs.setdefault('cmap', cmap)
kwargs.setdefault('vmin', lo)
@@ -160,8 +161,20 @@ def imshow(im, ax=None, show_cbar=None, **kwargs):
plt.colorbar(ax_im, cax=cax)
ax.set_adjustable('box-forced')
ax.get_figure().tight_layout()
return ax_im
def imshow_collection(ic, *args, **kwargs):
"""Display all images in the collection.
"""
fig, axes = plt.subplots(1, len(ic))
for n, image in enumerate(ic):
kwargs['axis'] = axes[n]
imshow(image, *args, **kwargs)
imread = plt.imread
show = plt.show