Exemplo n.º 1
0
    def read_cmap(self, cmap):
        """"""

        try:
            with open(cmap) as fid:
                rgb = [np.array(line.strip().lstrip('[').rstrip(']').strip().split(), dtype=float) for line in fid]
                cmap = ListedColormap(rgb)
        except IOError:
            try:
                cmap = getattr(plt.cm, cmap)
            except AttributeError:
                pkg, cmap = cmap.split('.')
                cmap = getattr(__import__(pkg), cmap)
            except:
                raise RuntimeError('Could not read colour map "{}"'.format(cmap))

        return cmap