Extends color list for cell plotting.
This commit is contained in:
@@ -121,9 +121,12 @@ if __name__ == "__main__":
|
||||
color_list = None
|
||||
|
||||
cmap = matplotlib.cm.get_cmap("Set3")
|
||||
unique_cells = transmission_df["cell_color"].unique()
|
||||
color_list = cmap * (round(len(unique_cells) / len(cmap)) + 1)
|
||||
|
||||
for c in transmission_df["cell_color"].unique():
|
||||
bounds = transmission_df[["index", "cell_color"]].groupby("cell_color").agg(["min", "max"]).loc[c]
|
||||
ax.axvspan(bounds.min(), bounds.max(), alpha=0.3, color=cmap.colors[c])
|
||||
ax.axvspan(bounds.min(), bounds.max(), alpha=0.3, color=color_list[c])
|
||||
|
||||
p4, = twin3.plot(transmission_df["snd_cwnd"].dropna(), color="lime", linestyle="dashed", label="cwnd")
|
||||
p3, = twin2.plot(transmission_df["ack_rtt"].dropna(), color="red", linestyle="dashdot", label="ACK RTT")
|
||||
|
||||
@@ -205,9 +205,12 @@ if __name__ == "__main__":
|
||||
color_list = None
|
||||
|
||||
cmap = matplotlib.cm.get_cmap("Set3")
|
||||
unique_cells = transmission_df["cell_color"].unique()
|
||||
color_list = cmap * (round(len(unique_cells) / len(cmap)) + 1)
|
||||
|
||||
for c in transmission_df["cell_color"].unique():
|
||||
bounds = transmission_df[["index", "cell_color"]].groupby("cell_color").agg(["min", "max"]).loc[c]
|
||||
ax.axvspan(bounds.min(), bounds.max(), alpha=0.3, color=cmap.colors[c])
|
||||
ax.axvspan(bounds.min(), bounds.max(), alpha=0.3, color=color_list[c])
|
||||
|
||||
p1, = ax.plot(transmission_df["goodput_rolling"], "-", color="blue", label="goodput")
|
||||
p2, = twin1.plot(transmission_df["downlink_cqi"], "--", color="green", label="CQI")
|
||||
|
||||
Reference in New Issue
Block a user