def boxplot(self, year, path=None): ''' This function creates a boxplot of income by region for the given year. The plot will be saved to path if specified, otherwise, it will be shown ''' mergedData = merge_by_year(year).dropna() plot = mergedData.boxplot(by="Region", rot=25).get_figure() if path == None: plot.show() else: plot.savefig(path)
def histogram(self, year, path=None): ''' This function creates a boxplot of income by region for the given year. The plot will be saved to path if specified, otherwise, it will be shown ''' mergedData=merge_by_year(year).dropna() mergedData.hist("Income", by="Region", bins=20, xrot=25, xlabelsize=8, figsize=(8,8)) if path==None: plt.show() else: plt.savefig(path)
def boxplot(self, year, path=None): ''' This function creates a boxplot of income by region for the given year. The plot will be saved to path if specified, otherwise, it will be shown ''' mergedData=merge_by_year(year).dropna() plot=mergedData.boxplot(by="Region", rot=25).get_figure() if path==None: plot.show() else: plot.savefig(path)
def histogram(self, year, path=None): ''' This function creates a boxplot of income by region for the given year. The plot will be saved to path if specified, otherwise, it will be shown ''' mergedData = merge_by_year(year).dropna() mergedData.hist("Income", by="Region", bins=20, xrot=25, xlabelsize=8, figsize=(8, 8)) if path == None: plt.show() else: plt.savefig(path)
def test_merge_region(self): self.assertEqual(merge_by_year(1901)[merge_by_year(1901)["Country"]=="United States"]["Region"].item(), countries[countries["Country"]=="United States"]["Region"].item()) self.assertEqual(merge_by_year(2001)[merge_by_year(2001)["Country"]=="United States"]["Region"].item(), merge_by_year(1901)[merge_by_year(1901)["Country"]=="United States"]["Region"].item()) self.assertEqual(merge_by_year(1852)[merge_by_year(1852)["Country"]=="Australia"]["Region"].item(), countries[countries["Country"]=="Australia"]["Region"].item()) self.assertEqual(merge_by_year(1985)[merge_by_year(1985)["Country"]=="China"]["Region"].item(), countries[countries["Country"]=="China"]["Region"].item())
def test_merge_income(self): self.assertEqual(merge_by_year(1901)[merge_by_year(1901)["Country"]=="United States"]["Income"].item(), income_transpose["United States"].loc[1901]) self.assertEqual(merge_by_year(2001)[merge_by_year(2001)["Country"]=="United States"]["Income"].item(), income_transpose["United States"].loc[2001]) self.assertEqual(merge_by_year(1852)[merge_by_year(1852)["Country"]=="Australia"]["Income"].item(), income_transpose["Australia"].loc[1852]) self.assertEqual(merge_by_year(1985)[merge_by_year(1985)["Country"]=="China"]["Income"].item(), income_transpose["China"].loc[1985])
def test_merge_columns(self): self.assertListEqual(list(merge_by_year(1801).columns.values),['Country', 'Income', 'Region'])