.. DO NOT EDIT.
.. THIS FILE WAS AUTOMATICALLY GENERATED BY SPHINX-GALLERY.
.. TO MAKE CHANGES, EDIT THE SOURCE PYTHON FILE:
.. "gallery/2D/09-plot2d_disc_cbar.py"
.. LINE NUMBERS ARE GIVEN BELOW.

.. only:: html

    .. note::
        :class: sphx-glr-download-link-note

        :ref:`Go to the end <sphx_glr_download_gallery_2D_09-plot2d_disc_cbar.py>`
        to download the full example code

.. rst-class:: sphx-glr-example-title

.. _sphx_glr_gallery_2D_09-plot2d_disc_cbar.py:


Display discrete colorbar, and set mask
===========================================

.. image:: ../../_static/cigvis/2D/09.png
    :alt: image
    :align: center

.. GENERATED FROM PYTHON SOURCE LINES 14-84

.. code-block:: default


    # sphinx_gallery_thumbnail_path = '_static/cigvis/2D/09.png'

    import numpy as np
    import cigvis
    from cigvis import colormap
    import matplotlib.pyplot as plt

    root = '../../data/'
    sxp = root + 'seis_h360x600x400.dat'
    lxp = root + 'label_h360x600x400.dat'
    ni, nx, nt = 400, 600, 360

    sx = np.memmap(sxp, np.float32, 'r', shape=(ni, nx, nt))
    lx = np.memmap(lxp, np.float32, 'r', shape=(ni, nx, nt))

    sx2 = sx[:, 100, :]
    lx2 = lx[:, 100, :]
    ticks_label = ['Class A', 'Class B', 'Class C', 'Class D', 'Class E']

    fig, axes = plt.subplots(2, 2, figsize=(9, 6.5))

    fg = {}
    fg['img'] = cigvis.fg_image_args(lx2, alpha=0.5, interpolation='nearest')
    cigvis.plot2d(sx2,
                  fg,
                  cbar='Facies',
                  discrete=True,
                  tick_labels=ticks_label,
                  title='discrete colorbar',
                  ax=axes[0, 0])

    cmap = colormap.set_alpha_except_min('jet', 0.5, False)
    fg['img'] = cigvis.fg_image_args(lx2, cmap=cmap, interpolation='nearest')
    cigvis.plot2d(sx2,
                  fg,
                  cbar='Facies',
                  discrete=True,
                  tick_labels=ticks_label,
                  title='remove the min value',
                  ax=axes[0, 1])

    cmap = colormap.set_alpha_except_min('jet', 0.5, False)
    fg['img'] = cigvis.fg_image_args(lx2, cmap=cmap, interpolation='nearest')
    cigvis.plot2d(sx2,
                  fg,
                  cbar='Facies',
                  discrete=True,
                  tick_labels=ticks_label,
                  remove_trans=False,
                  title='cbar keep the min value',
                  ax=axes[1, 0])

    colors = ['red', 'green', 'yellow', 'blue', (0, 0.5, 0.5)]
    values = np.unique(lx2)
    cmap = colormap.custom_disc_cmap(values, colors)
    cmap = colormap.set_alpha_except_min(cmap, 0.5, False)
    fg['img'] = cigvis.fg_image_args(lx2, cmap=cmap, interpolation='nearest')
    cigvis.plot2d(sx2,
                  fg,
                  cbar='Facies',
                  discrete=True,
                  tick_labels=ticks_label,
                  title='custom cmap',
                  ax=axes[1, 1])

    plt.tight_layout()
    plt.savefig('example.png', bbox_inches='tight', pad_inches=0.01, dpi=200)

    plt.show()


.. rst-class:: sphx-glr-timing

   **Total running time of the script:** (0 minutes 0.000 seconds)


.. _sphx_glr_download_gallery_2D_09-plot2d_disc_cbar.py:

.. only:: html

  .. container:: sphx-glr-footer sphx-glr-footer-example




    .. container:: sphx-glr-download sphx-glr-download-python

      :download:`Download Python source code: 09-plot2d_disc_cbar.py <09-plot2d_disc_cbar.py>`

    .. container:: sphx-glr-download sphx-glr-download-jupyter

      :download:`Download Jupyter notebook: 09-plot2d_disc_cbar.ipynb <09-plot2d_disc_cbar.ipynb>`


.. only:: html

 .. rst-class:: sphx-glr-signature

    `Gallery generated by Sphinx-Gallery <https://sphinx-gallery.github.io>`_