def weekly_increase()

in api/streamlit_experiments/covid.py [0:0]

• 30 lines of code
• 2 McCabe index (conditional complexity)

def weekly_increase(df):
    week_num=[]
    weekwise_confirmed=[]
    weekwise_recovered=[]
    weekwise_deaths=[]
    w=1
    for i in list(df["WeekOfYear"].unique()):
        weekwise_confirmed.append(df[df["WeekOfYear"]==i]["Confirmed"].iloc[-1])
        weekwise_recovered.append(df[df["WeekOfYear"]==i]["Recovered"].iloc[-1])
        weekwise_deaths.append(df[df["WeekOfYear"]==i]["Deaths"].iloc[-1])
        week_num.append(w)
        w=w+1

    fig = plt.figure(figsize=(8,5))
    plt.plot(week_num,weekwise_confirmed,linewidth=3)
    plt.plot(week_num,weekwise_recovered,linewidth=3)
    plt.plot(week_num,weekwise_deaths,linewidth=3)
    plt.ylabel("Number of Cases")
    plt.xlabel("Week Number")
    plt.title("Weekly progress of Different Types of Cases")
    # plt.xlabel
    st.pyplot(fig)

    fig, (ax1,ax2) = plt.subplots(1, 2,figsize=(15,5))
    sns.barplot(x=week_num,y=pd.Series(weekwise_confirmed).diff().fillna(0),ax=ax1)
    sns.barplot(x=week_num,y=pd.Series(weekwise_deaths).diff().fillna(0),ax=ax2)
    ax1.set_xlabel("Week Number")
    ax2.set_xlabel("Week Number")
    ax1.set_ylabel("Number of Confirmed Cases")
    ax2.set_ylabel("Number of Death Cases")
    ax1.set_title("Weekly increase in Number of Confirmed Cases")
    ax2.set_title("Weekly increase in Number of Death Cases")

    st.pyplot(fig)