stack/infra/marimo/snippets/query-pmpm.py

import marimo

__generated_with = "0.20.4"
app = marimo.App()


@app.cell
def _(mo):
    mo.md(r"""
    # Query: PMPM financial summary

    Per-member-per-month cost breakdown by service category.
    """)
    return


@app.cell
def _(mo, q):
    pmpm = q('''
        SELECT year_month, member_months,
               round(total_paid / member_months, 2) AS total_pmpm,
               round(inpatient_paid / member_months, 2) AS inpatient_pmpm,
               round(pharmacy_paid / member_months, 2) AS pharmacy_pmpm
        FROM financial_pmpm.pmpm_payer
        WHERE member_months > 50
        ORDER BY year_month
    ''')
    mo.ui.table(pmpm, label="PMPM by Month")
    return (pmpm,)


@app.cell
def _():
    import marimo as mo

    return (mo,)


if __name__ == "__main__":
    app.run()