From 213e04a1e0079611d297554e61922baff61ba21c Mon Sep 17 00:00:00 2001 From: oldoc63 Date: Sun, 17 Oct 2021 13:45:32 -0400 Subject: [PATCH] Splitting by GDP #149 --- life/script.py | 10 ++++++++++ 1 file changed, 10 insertions(+) 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()