langspielplatte/measurement-scripts
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Background color for cell id.

Browse Source
This commit is contained in:
Lukas Prause
2023-01-19 14:24:04 +01:00
parent f0df8aeb2c
commit b55d38c5b9
1 changed files with 10 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
plot_transmission_timeline.py
View File

@@ -189,12 +189,19 @@ if __name__ == "__main__":
host = host_subplot(111, axes_class=axisartist.Axes)
# create list fo color indices
transmission_df["index"] = transmission_df.index
color_dict = dict()
i = 0
for cell_id in transmission_df["cellID"].unique():
if cell_id not in color_dict:
color_dict[cell_id] = i
i += 1
cmap = matplotlib.cm.get_cmap("Set3")
for c in transmission_df["cellID"].unique():
bounds = transmission_df[["index", "cellID"]].groupby("cellID").agg(["min", "max"]).loc[c]
host.axvspan(bounds.min(), bounds.max(), alpha=0.3, color=cmap.colors[c])
for cell_id in transmission_df["cellID"].unique():
bounds = transmission_df[["index", "cellID"]].groupby("cellID").agg(["min", "max"]).loc[cell_id]
host.axvspan(bounds.min(), bounds.max(), alpha=0.3, color=cmap.colors[color_dict[cell_id]])
plt.subplots_adjust()