12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849 |
- 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)
|