diff --git a/life/script.py b/life/script.py index 7d17fbb..92ea04c 100644 --- a/life/script.py +++ b/life/script.py @@ -8,10 +8,20 @@ print(data.head()) life_expectancy = data["Life Expectancy"] +gdp = data["GDP"] life_expectancy_quartiles = np.quantile(life_expectancy, [0.25, 0.50, 0.75]) +median_gdp = np.quantile(gdp, 0.5) +low_gdp = data[data['GDP'] <= median_gdp] +high_gdp = data[data['GDP'] > median_gdp] + +low_gdp_quartiles = np.quantile(low_gdp["Life Expectancy"], [0.25, 0.50, 0.75]) +high_gdp_quartiles = np.quantile(high_gdp["Life Expectancy"], [0.25, 0.50, 0.75]) + print(life_expectancy_quartiles) +print(low_gdp_quartiles) +print(high_gdp_quartiles) plt.hist(life_expectancy) plt.show()