save_trendline.py

#!/usr/bin/env python
# coding: utf-8

# CDC: Monkeypox visualizations

import pandas as pd
import altair as alt
import altair_grid as altgrid
from altair_saver import save

alt.themes.register("grid", altgrid.theme)
alt.themes.enable("grid")
pd.options.display.max_columns = 1000
pd.options.display.max_rows = 1000
alt.data_transformers.disable_max_rows()
alt.renderers.enable("altair_saver", fmts=["vega-lite", "png"])

today = pd.Timestamp.now(tz="America/Los_Angeles").strftime("%Y-%m-%d")
time = pd.Timestamp.now(tz="America/Los_Angeles").strftime("%-I:%M %p")

## Cases timeseries

src = pd.read_csv(
    "data/processed/monkeypox_cases_derived_timeseries_latest.csv"
)

src.head()

line = (
    alt.Chart(src)
    .mark_line()
    .encode(
        x=alt.X("date:T", axis=alt.Axis(tickCount=5), title=""),
        y=alt.Y("cumulative_sum:Q", title=" ", axis=alt.Axis(tickCount=5)),
    )
    .properties(title="Cumulative monkeypox cases in the U.S.", width=650, height=300)
)

text = (
    alt.Chart(src.query("(cumulative_sum == cumulative_sum.max()) & (cases > 0)"))
    .mark_text(dy=-10, dx=10)
    .encode(
        x=alt.X("date:T"),
        y=alt.Y("cumulative_sum:Q"),
        text=alt.Text("cumulative_sum:Q"),
    )
)

circle = (
    alt.Chart(src.query("(date == date.max())"))
    .mark_circle(color="#00d4d8", size=50)
    .encode(
        x=alt.X("date:T"),
        y=alt.Y("cumulative_sum:Q"),
    )
)

chart = (line + text + circle).configure_legend(symbolType="stroke", orient="top")

chart.save("visuals/trendline_latest.png")
chart.save("visuals/trendline.png")
chart.save(f"visuals/trendline_{today}.png")