1234567891011121314151617181920212223242526272829303132333435363738394041 |
- 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()
|