covid_analisis_jupyter_04.py

# Hacemos que la columna Entity sea un índice
f_2 = filtered.set_index('Entity')
# Volvemos a filtrar, pero ahora los países potencia junto con México
power_countries = f_2.loc[ ['Mexico', 'United States' , 
                            'United Kingdom', 'Russia', 
                            'South Korea'] ]

#Creamos una Pivot table que sume todas las pruebas de cada país
table = pd.pivot_table(
    power_countries,
    values = 'New tests',
    index = ['Entity'],
    aggfunc=sum,
    fill_value=0
).sort_values(by='New tests', ascending=True)

#Cambiamos el nombre de la columna New test a Pruebas
table = table.rename(columns={ 'New tests':'Pruebas' })
table