structured_inhibition_spatial_network_model/scripts/model_figure/max_entropy_bar_plots.py

import matplotlib.pyplot as plt
from matplotlib import cm
import numpy as np

FIGURE_SAVE_PATH = '../../figures/figure_4_paper_draft/'


cmap = cm.get_cmap('hsv')

high_entropy = [2, 2, 1, 1]
low_entropy = [4, 2, 0, 0]


bars = ('0°', '90°', '180°', '270°')
xticks = [0, 90, 180, 270]
y_pos = np.arange(len(bars))

ax = plt.subplot(111)

ax.spines['right'].set_visible(False)
ax.spines['top'].set_visible(False)

ax.bar(y_pos, low_entropy, color=[cmap(0.5), cmap(0.75), cmap(0.), cmap(0.25)])
ax.set_xticks(y_pos)
ax.set_xticklabels(bars, fontsize=20)
ax.set_yticks([])
ax.set_ylim(0,4)

for axis in ['bottom','left']:
  ax.spines[axis].set_linewidth(2.5)

plt.savefig(FIGURE_SAVE_PATH + 'low_entropy_bar_plot.png', dpi=200)
plt.close()

ax2 = plt.subplot(111)

ax2.spines['right'].set_visible(False)
ax2.spines['top'].set_visible(False)

ax2.bar(y_pos, high_entropy, color=[cmap(0.5), cmap(0.75), cmap(0.), cmap(0.25)])
ax2.set_xticks(y_pos)
ax2.set_xticklabels(bars, fontsize=20)
ax2.set_yticks([])
ax2.set_ylim(0,4)

for axis in ['bottom','left']:
  ax2.spines[axis].set_linewidth(2.5)

plt.savefig(FIGURE_SAVE_PATH + 'high_entropy_bar_plot.png', dpi=200)