Background color for cell id.

This commit is contained in:
Lukas Prause
2023-01-19 14:26:47 +01:00
parent b55d38c5b9
commit 675764bdc6

View File

@@ -199,9 +199,9 @@ if __name__ == "__main__":
i += 1 i += 1
cmap = matplotlib.cm.get_cmap("Set3") cmap = matplotlib.cm.get_cmap("Set3")
for cell_id in transmission_df["cellID"].unique(): for c in transmission_df["cellID"].unique():
bounds = transmission_df[["index", "cellID"]].groupby("cellID").agg(["min", "max"]).loc[cell_id] bounds = transmission_df[["index", "cellID"]].groupby("cellID").agg(["min", "max"]).loc[c]
host.axvspan(bounds.min(), bounds.max(), alpha=0.3, color=cmap.colors[color_dict[cell_id]]) host.axvspan(bounds.min(), bounds.max(), alpha=0.3, color=cmap.colors[color_dict[c]])
plt.subplots_adjust() plt.subplots_adjust()