From b7676674eec5b627eff6d177f273d845adc85806 Mon Sep 17 00:00:00 2001 From: Lukas Prause Date: Thu, 19 Jan 2023 13:00:55 +0100 Subject: [PATCH] Background color for cell id. --- plot_transmission_timeline.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/plot_transmission_timeline.py b/plot_transmission_timeline.py index 54405c4..f13ea92 100755 --- a/plot_transmission_timeline.py +++ b/plot_transmission_timeline.py @@ -189,11 +189,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[["cellID"]].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[[transmission_df.index, "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()