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:39:03 +01:00
parent 675764bdc6
commit 64360e9a28
1 changed files with 10 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
plot_transmission_timeline.py
View File

@@ -192,16 +192,22 @@ if __name__ == "__main__":
# create list fo color indices # create list fo color indices
transmission_df["index"] = transmission_df.index transmission_df["index"] = transmission_df.index
color_dict = dict() color_dict = dict()
color_list = list()
i = 0 i = 0
for cell_id in transmission_df["cellID"].unique(): for cell_id in transmission_df["cellID"]:
if cell_id not in color_dict: if cell_id not in color_dict:
color_dict[cell_id] = i color_dict[cell_id] = i
i += 1 i += 1
color_list.append(color_dict[cell_id])
transmission_df["cell_color"] = color_list
color_dict = None
color_list = None
cmap = matplotlib.cm.get_cmap("Set3") cmap = matplotlib.cm.get_cmap("Set3")
for c in transmission_df["cellID"].unique(): for c in transmission_df["cell_color"].unique():
bounds = transmission_df[["index", "cellID"]].groupby("cellID").agg(["min", "max"]).loc[c] bounds = transmission_df[["index", "cell_color"]].groupby("cell_color").agg(["min", "max"]).loc[c]
host.axvspan(bounds.min(), bounds.max(), alpha=0.3, color=cmap.colors[color_dict[c]]) host.axvspan(bounds.min(), bounds.max(), alpha=0.3, color=cmap.colors[c])
plt.subplots_adjust() plt.subplots_adjust()