FIX: FutureWarning in heavy_tails

Author: mmcky

lectures/heavy_tails.md

@@ -4,7 +4,7 @@ jupytext:
     extension: .md
     format_name: myst
     format_version: 0.13
-    jupytext_version: 1.16.1
+    jupytext_version: 1.16.7
 kernelspec:
   display_name: Python 3 (ipykernel)
   language: python
@@ -31,7 +31,7 @@ import yfinance as yf
 import pandas as pd
 import statsmodels.api as sm
 
-from pandas_datareader import wb
+import wbgapi as wb
 from scipy.stats import norm, cauchy
 from pandas.plotting import register_matplotlib_converters
 register_matplotlib_converters()
@@ -790,24 +790,24 @@ def empirical_ccdf(data,
 :tags: [hide-input]
 
 def extract_wb(varlist=['NY.GDP.MKTP.CD'], 
-               c='all_countries', 
+               c='all', 
                s=1900, 
                e=2021, 
                varnames=None):
-    if c == "all_countries":
-        # Keep countries only (no aggregated regions)
-        countries = wb.get_countries()
-        countries_name = countries[countries['region'] != 'Aggregates']['name'].values
-        c = "all"
     
-    df = wb.download(indicator=varlist, country=c, start=s, end=e).stack().unstack(0).reset_index()
-    df = df.drop(['level_1'], axis=1).transpose()
+    df = wb.data.DataFrame(varlist, economy=c, time=range(s, e+1, 1), skipAggs=True)
+    df.index.name = 'country'
+    
     if varnames is not None:
-        df.columns = varnames
-        df = df[1:]
+        df.columns = variable_names
+        
+    df.index = df.index.map(lambda x: cntry_mapper[x])  #map iso3c to name values
     
-    df1 =df[df.index.isin(countries_name)]
-    return df1
+    return df
+```
+
+```{code-cell} ipython3
+df_gdp1
 ```
 
 ### Firm size
@@ -914,13 +914,25 @@ variable_code = ['NY.GDP.MKTP.CD', 'NY.GDP.PCAP.CD']
 variable_names = ['GDP', 'GDP per capita']
 
 df_gdp1 = extract_wb(varlist=variable_code, 
-                     c="all_countries", 
-                     s="2021", 
-                     e="2021", 
+                     c="all", 
+                     s=2021, 
+                     e=2021, 
                      varnames=variable_names)
 df_gdp1.dropna(inplace=True)
 ```
 
+```{code-cell} ipython3
+
+```
+
+```{code-cell} ipython3
+
+```
+
+```{code-cell} ipython3
+
+```
+
 ```{code-cell} ipython3
 ---
 mystnb: