bsta040test/project.py at main · computationalUncertaintyLab/bsta040test · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
#mcandrew

import sys
import numpy as np
import pandas as pd

import matplotlib.pyplot as plt

import streamlit as st
import scipy

if __name__ == "__main__":

    ili_data = pd.read_csv("./ilidata.csv")

    ili_data["year"] = ili_data["epiweek"].astype(str).str[:4]
    ili_data["week"] = ili_data["epiweek"].astype(str).str[-2:]

    def pick_state(state):
        pa = ili_data.loc[ili_data.state==state]
        weeks                   = np.arange(len(pa))
        pa["weeks"] = weeks

        return pa

    option = st.selectbox("Select State", ili_data.state.unique() )

    #option = "pa"
    all_state_data = pick_state(option)

    st.line_chart(data = all_state_data, x="weeks",y=["ili"])


    fig,ax = plt.subplots()
    plt.hist( all_state_data.ili.values, density=True, bins=20)
    est_rate = 1./np.mean(all_state_data.ili)

    dom  = np.linspace(0.05,7,50)
    cdfs = scipy.stats.expon( 0, est_rate ).pdf(dom)

    plt.plot(dom,cdfs)

    st.pyplot( fig )