structured_inhibition_spatial_network_model/scripts/model_figure/plot_circular_colorbar.py

import matplotlib.pyplot as plt
import numpy as np

from scripts.spatial_network.perlin.figure_utils import head_direction_input_colormap

azimuth_no = 360
zenith_no = 15

azimuths = np.linspace(-180, 180, azimuth_no)
zeniths = np.linspace(0.85, 1, zenith_no)

values = azimuths * np.ones((zenith_no, azimuth_no))


def plot_colorbar(figsize=(2, 2), figname=None):
    fig, ax = plt.subplots(subplot_kw=dict(projection='polar'), figsize=figsize)
    ax.pcolormesh(azimuths * np.pi / 180.0, zeniths, values, cmap=head_direction_input_colormap)
    # ax.set_yticks([])
    ax.set_thetagrids([0, 90, 180, 270])
    ax.tick_params(pad=-2)
    ax.set_ylim(0, 1)
    ax.grid(True)

    y_tick_labels = []
    ax.set_yticklabels(y_tick_labels)

    gridlines = ax.yaxis.get_gridlines()
    [line.set_linewidth(0.0) for line in gridlines]

    gridlines = ax.xaxis.get_gridlines()
    [line.set_linewidth(0.0) for line in gridlines]
    ax.axes.spines["polar"].set_visible(False)
    plt.subplots_adjust(left=0.25, right=0.75, bottom=0.25, top=0.75)
    # plt.show()
    if figname is not None:
        plt.savefig(figname, transparent=True)


if __name__ == "__main__":
    plot_colorbar()
    plt.show()