.. DO NOT EDIT.
.. THIS FILE WAS AUTOMATICALLY GENERATED BY SPHINX-GALLERY.
.. TO MAKE CHANGES, EDIT THE SOURCE PYTHON FILE:
.. "plot_types/gridded/heatmap.py"
.. LINE NUMBERS ARE GIVEN BELOW.

.. only:: html

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

        :ref:`Go to the end <sphx_glr_download_plot_types_gridded_heatmap.py>`
        to download the full example code.

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

.. _sphx_glr_plot_types_gridded_heatmap.py:


Heatmap
-------

Below is an example of how to plot a categorical heatmap
using EMCPy's plotting method.

.. GENERATED FROM PYTHON SOURCE LINES 9-56



.. image-sg:: /plot_types/gridded/images/sphx_glr_heatmap_001.png
   :alt: Heatmap
   :srcset: /plot_types/gridded/images/sphx_glr_heatmap_001.png
   :class: sphx-glr-single-img





.. code-block:: Python


    import numpy as np
    import matplotlib.pyplot as plt

    from emcpy.plots.plots import HeatMap
    from emcpy.plots.create_plots import CreatePlot, CreateFigure


    def main():
        # Create heatmap plot

        # Grab sample data
        x, y, data = _getHeatmapData()

        # Create heatmap object
        hm = HeatMap(x, y, data)
        hm.cmap = 'viridis'

        # Create plot object and add features
        plot1 = CreatePlot()
        plot1.plot_layers = [hm]
        plot1.add_xlabel(xlabel='X Axis Label')
        plot1.add_ylabel(ylabel='Y Axis Label')
        plot1.add_title('Heatmap')
        plot1.add_colorbar(orientation='vertical')

        # Create figure
        fig = CreateFigure()
        fig.plot_list = [plot1]
        fig.create_figure()

        plt.show()


    def _getHeatmapData():
        # Generate test data for heatmap

        x = [f'Col {i}' for i in range(6)]
        y = [f'Row {i}' for i in range(4)]
        r = np.random.RandomState(25)
        data = r.random_sample([len(y), len(x)])

        return x, y, data


    if __name__ == '__main__':
        main()


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

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


.. _sphx_glr_download_plot_types_gridded_heatmap.py:

.. only:: html

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

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

      :download:`Download Jupyter notebook: heatmap.ipynb <heatmap.ipynb>`

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

      :download:`Download Python source code: heatmap.py <heatmap.py>`

    .. container:: sphx-glr-download sphx-glr-download-zip

      :download:`Download zipped: heatmap.zip <heatmap.zip>`


.. only:: html

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

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