I am trying to create a polar chart that has four categories. Each category has two values: sleep_time and ahi. I am wanting the size of each bar to be proportional to the sleep_time - which seems to be working. I also want the color to be based on the ahi that is also associated with each category. Then have a color bar to reflect this. I can get the sleep_time function going but my color is the issue. This is what I have so far: import matplotlib.pyplot as plt import numpy as np categories = ['Right', 'Supine', 'Left', 'Prone'] sleep_time = [251.5, 40, 130, 15.5] ahi = [12.5, 65.6, 8.2, 0.0] angles = np.linspace(0, 2*np.pi, len(categories), endpoint=False) fig, ax = plt.subplots(figsize=(8, 8), subplot_kw=dict(projection='polar')) colors = plt.cm.viridis_r(np.linspace(0, 100, len(ahi))) ax.bar(angles, sleep_time, width=1.0, align='center', alpha=0.5) ax.bar(angles, ahi, width=0.4, align='edge', alpha=0.5, color=colors) ax.set_xticks(angles) ax.set_xticklabels(categories) colorbar = plt.colorbar(ahi) colorbar.set_ticks([0, 30, 60, 90, 120, 150, 180]) plt.show() plt.savefig('position_vs._ahi.png') Continue reading...
