diff --git a/plot_transmission_timeline.py b/plot_transmission_timeline.py index 66b22d3..2429bbb 100755 --- a/plot_transmission_timeline.py +++ b/plot_transmission_timeline.py @@ -185,10 +185,10 @@ if __name__ == "__main__": host = host_subplot(111, axes_class=axisartist.Axes) - cmap = matplotlib.cm.get_cmap("Set3") - for c in transmission_df["cellID"].unique(): - bounds = transmission_df.groupby("cellID").agg(["min", "max"]).loc[c] - host.axvspan(bounds.min(), bounds.max() + 1, alpha=0.3, color=cmap.colors[c]) + #cmap = matplotlib.cm.get_cmap("Set3") + #for c in transmission_df["cellID"].unique(): + # bounds = transmission_df[["cellID"]].groupby("cellID").agg(["min", "max"]).loc[c] + # host.axvspan(bounds.min(), bounds.max() + 1, alpha=0.3, color=cmap.colors[c]) plt.subplots_adjust()