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

Extends color list for cell plotting.

Browse Source
This commit is contained in:
Lukas Prause
2023-01-23 13:16:09 +01:00
parent 1b97e12b14
commit d3880d4ffc
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
plot_single_transmission_timeline.py
View File

@@ -122,7 +122,7 @@ if __name__ == "__main__":
cmap = matplotlib.cm.get_cmap("Set3") cmap = matplotlib.cm.get_cmap("Set3")
unique_cells = transmission_df["cell_color"].unique() unique_cells = transmission_df["cell_color"].unique()
color_list = cmap * (round(len(unique_cells) / len(cmap)) + 1) color_list = cmap.colors * (round(len(unique_cells) / len(cmap.colors)) + 1)
for c in transmission_df["cell_color"].unique(): for c in transmission_df["cell_color"].unique():
bounds = transmission_df[["index", "cell_color"]].groupby("cell_color").agg(["min", "max"]).loc[c] bounds = transmission_df[["index", "cell_color"]].groupby("cell_color").agg(["min", "max"]).loc[c]

2
plot_transmission_timeline.py
View File

@@ -206,7 +206,7 @@ if __name__ == "__main__":
cmap = matplotlib.cm.get_cmap("Set3") cmap = matplotlib.cm.get_cmap("Set3")
unique_cells = transmission_df["cell_color"].unique() unique_cells = transmission_df["cell_color"].unique()
color_list = cmap * (round(len(unique_cells) / len(cmap)) + 1) color_list = cmap.colors * (round(len(unique_cells) / len(cmap.colors)) + 1)
for c in transmission_df["cell_color"].unique(): for c in transmission_df["cell_color"].unique():
bounds = transmission_df[["index", "cell_color"]].groupby("cell_color").agg(["min", "max"]).loc[c] bounds = transmission_df[["index", "cell_color"]].groupby("cell_color").agg(["min", "max"]).loc[c]