From 18300b800cf990f007b773323333ca4e509e7361 Mon Sep 17 00:00:00 2001 From: rabons Date: Wed, 30 Mar 2016 17:44:38 -0400 Subject: [PATCH] Adrien's changes to lrv_or_mrv --- wb_api_wrapper.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/wb_api_wrapper.py b/wb_api_wrapper.py index 4a064a4..aaea3ce 100644 --- a/wb_api_wrapper.py +++ b/wb_api_wrapper.py @@ -73,14 +73,14 @@ def lrv_or_mrv(data,least_or_most,include_year=False): data = data.unstack() except IndexError : pass #data is already a series; - + #removes nans, and takes the least or most recent value. hop has a horrible shape hop=data.reset_index().dropna().groupby("country").apply(mlrv_gp,least_or_most,include_year) if include_year: return pd.DataFrame(hop.tolist(), columns=['value',"year"], index=hop.index) else: - return hop.reset_index().set_index("country")[0] + return hop.reset_index().set_index("country")[data.columns[0]] def mr_year(data): """year of most recent values from a dataframe. assumes one column is called 'year'