# Prog-07: COVID-19
# Fill in your ID & Name
# ...
# Declare that you do this by yourself

import pandas as pd

def get_data():
    return pd.read_pickle('covid19_df.pickle')
    csv = 'https://raw.githubusercontent.com/CSSEGISandData/COVID-19/' + \
          'master/csse_covid_19_data/csse_covid_19_time_series/' + \
          'time_series_covid19_confirmed_global.csv'
    df=pd.read_csv(csv)
    df=df.groupby('Country/Region').sum()
    df=df.T.iloc[2:]
    return df

def get_confirmed_cases(df,country):
    return list(df[country])

def get_all_countries(df):
    return list(df.columns)
#-------------------------------------------------

def steepest_countries(df, t, n):



def most_similar_countries(df, country, t, n):
    


def top_k_similar_countries(df, country, t, n, k):



#-------------------------------------------------
def main():
    df=get_data()
    print(steepest_countries(df, 1000000, 10))
    print(steepest_countries(df, 10000, 10))
    print(steepest_countries(df, 100, 10))
    print(most_similar_countries(df, 'Burma', 1000, 7))
    print(top_k_similar_countries(df, 'Cyprus', 1055, 5, 4))
#-------------------------------------------------
main()