fblanke/tvstud-download-digital
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update plot

Browse Source
This commit is contained in:
Felix Blanke 2023-09-28 20:38:48 +02:00
parent 6060930208
commit d205625ef4
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
wsgi.py
View File

@ -113,7 +113,10 @@ def plot(
fix_lims: bool = True,
max_shading_date=None,
) -> str:
fig = plt.figure(dpi=300, figsize=(7, 5))
fig = plt.figure(dpi=300, figsize=(8.5, 5))
target_time = pd.Timestamp("2023-10-01")
plt.axvline(x=target_time, color="tab:green", linestyle=":")
if fix_lims:
for total_target in total_targets:
@ -202,14 +205,14 @@ def plot(
# fill weekends
if max_shading_date is None:
max_shading_date = df.index.max() + datetime.timedelta(days=3)
max_shading_date = df.index.max() + datetime.timedelta(days=4)
days = pd.date_range(start="2023-08-14", end=max_shading_date)
for idx, day in enumerate(days[:-1]):
if day.weekday() >= 5:
plt.gca().axvspan(days[idx], days[idx + 1], alpha=0.2, color="gray")
# reset xlim
plt.xlim(xlim)
plt.xlim((xlim[0], pd.Timestamp("2023-10-02")))
plt.tight_layout()