requesting API with multiple columns #5

Activity

1. neilsummers commented on Jul 1, 2021

   

   neilsummers

   on Jul 1, 2021

   More actions

   Here is the modified version I am using to allow pulling data with multiple columns.
   This does not iterate over rows so should be faster.
   Note I have also changed the column names to only include metric and not indicator.metric.

           try:
               df = pd.DataFrame(json.loads(r.text))
               df = df.set_index('t')
               df.index = pd.to_datetime(df.index, unit='s')
               df = df.sort_index()
               if 'v' in df:
                   s = df.v
                   # s.name = '_'.join(url.split('/')[-2:])
                   s.name = url.split('/')[-1]
                   return pd.DataFrame(s)
               else:  # 'o' contains json object of multiple columns
                   index = df.index
                   df = pd.json_normalize(df.o)
                   df.index = index
                   col_name_prefix = url.split('/')[-1]
                   df.columns = [f'{col_name_prefix}.{col}' for col in df.columns]
                   return df
           except Exception as e:
               print(e)