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

.. only:: html

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

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

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

.. _sphx_glr_auto_examples_plot_cord19.py:


CORD-19
-------

CORD-19 Data map with highlights applied to some important labels

.. GENERATED FROM PYTHON SOURCE LINES 7-55



.. image-sg:: /auto_examples/images/sphx_glr_plot_cord19_001.png
   :alt: CORD-19 Data Map, A data map of papers relating to COVID-19 and SARS-CoV-2
   :srcset: /auto_examples/images/sphx_glr_plot_cord19_001.png
   :class: sphx-glr-single-img





.. code-block:: Python

    import datamapplot
    import numpy as np
    import requests
    import PIL
    import matplotlib.pyplot as plt
    import pandas as pd

    plt.rcParams['savefig.bbox'] = 'tight'

    cord19_data_map = np.load("CORD19-subset-data-map.npz")["arr_0"]
    cord19_labels = np.load("CORD19-subset-cluster_labels.npz", allow_pickle=True)["arr_0"]

    # Prune labels down slightly
    label_counts = pd.Series(cord19_labels).value_counts()
    small_clusters = label_counts[label_counts <= 700].index
    for label in small_clusters:
        cord19_labels[cord19_labels == label] = "Unlabelled"

    allenai_logo_response = requests.get(
        "https://allenai.org/newsletters/archive/2023-03-newsletter_files/927c3ca8-6c75-862c-ee5d-81703ef10a8d.png",
        stream=True,
    )
    allenai_logo = np.asarray(PIL.Image.open(allenai_logo_response.raw).convert("RGBA"))

    fig, ax = datamapplot.create_plot(
        cord19_data_map,
        cord19_labels,
        palette_hue_shift=-90,
        title="CORD-19 Data Map",
        sub_title="A data map of papers relating to COVID-19 and SARS-CoV-2",
        highlight_labels=[
            "Effects of the COVID-19 pandemic on mental health",
            "Airborne Transmission of COVID19",
            "COVID19 Diagnosis",
            "Viral Diseases and Emerging Zoonoses",
            "Vaccine Acceptance",
        ],
        font_family="Cinzel",
        label_font_size=7,
        label_linespacing=1.25,
        label_margin_factor=1.5,
        label_direction_bias=1.0,
        highlight_label_keywords={"fontsize": 11, "fontweight": "bold", "bbox": {"boxstyle": "circle", "pad": 0.75}},
        title_keywords={"fontsize": 28},
        logo=allenai_logo,
    )
    fig.savefig("plot_cord19.png", bbox_inches="tight")
    plt.show()


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

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


.. _sphx_glr_download_auto_examples_plot_cord19.py:

.. only:: html

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

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

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

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

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

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

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


.. only:: html

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

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