def get_season(season, location): #end = (season + 1) * 100 + 29 #epiweeks = Epidata.range(flu.add_epiweeks(end, -51), end) begin = season * 100 + 30 epiweeks = Epidata.range(begin, flu.add_epiweeks(begin, 51)) rows = AF_Utils._get(Epidata.ilinet(location, epiweeks)) return [row['wili'] for row in rows]
def get_ilinet_data(states, start, end): """ return a dictionary of dataframe with the different epiweeks """ ilinet_raw = {} for state in states: print("State {}".format(state)) res = Epidata.ilinet( locations=state, #source epiweeks=[Epidata.range(start, end)]) #range 2009 to 2016 if res['result'] == 1: print(res['result'], res['message'], len(res['epidata'])) data = pd.DataFrame(res['epidata']) ilinet_raw[state] = data else: print("(-2, u'no success')") return ilinet_raw