Background color for cell id.

This commit is contained in:
Lukas Prause
2023-01-19 14:08:59 +01:00
parent b7676674ee
commit 84546710ed

View File

@@ -189,9 +189,11 @@ if __name__ == "__main__":
host = host_subplot(111, axes_class=axisartist.Axes) host = host_subplot(111, axes_class=axisartist.Axes)
transmission_df["index"] = transmission_df.index
cmap = matplotlib.cm.get_cmap("Set3") cmap = matplotlib.cm.get_cmap("Set3")
for c in transmission_df["cellID"].unique(): for c in transmission_df["cellID"].unique():
bounds = transmission_df[[transmission_df.index, "cellID"]].groupby("cellID").agg(["min", "max"]).loc[c] bounds = 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]) host.axvspan(bounds.min(), bounds.max() + 1, alpha=0.3, color=cmap.colors[c])
plt.subplots_adjust() plt.subplots_adjust()