dsfs 03 viz

matplotlib

1. simple plot line chart

        from matplotlib import pyplot as plt
   
        years = [1950, 1960, 1970, 1980, 1990, 2000, 2010]
        gdp = [300.2, 543.3, 1075.9, 2862.5, 5979.6, 102897, 14958.3]
   
        # create a line chart, years on x-axis, gdp on y-axis
        plt.plot(years, gdp, color='green', marker='o', linestyle='solid')
   
        # add a title
        plt.title('nominal gdp')
   
        # add a label to the y-axis
        plt.ylabel('billions of $')
        plt.show()
   

2. bar chart

        movies = ['annie hall', 'ben-hur', 'casablanca', 'gandi', 'west side story']
        num_oscars = [5, 11, 3, 8, 10]
   
        # bars are by default width 0.8
        # add 0.1 to the left coordinates
        xs = [i + 0.1 for i, _ in enumerate(movies)]
   
        # plot bars with left x-coordinates [xs] heights [num_oscars]
        plt.bar(xs, num_oscars)
   
        plt.ylabel('# of academy awards')
        plt.title('my favorite movies')
   
        # label x-axis with movie names at bar centers
        plt.xticks([i + 0.5 for i, _ in enumerate(movies)], movies)
   
        plt.show()