#! /usr/bin/env python3
#
def code02 ( ):

#*****************************************************************************80
#
## code02() reads price data and plots 3 prices on one display.
#
#  Discussion:
#
#    The file price_data.txt contains average monthly prices for 11
#    consumer products, between February 2008 and February 2018.
#
#    There are 241 records, each containing 13 items:
#    month, year, 11 prices
#
#    Columns 3, 10 and 13 contain the prices of bananas, gas, and milk.
#    We want a single plot that shows the variation in these three prices.
#
#  Licensing:
#
#    This code is distributed under the GNU LGPL license.
#
#  Modified:
#
#    26 August 2021
#
#  Author:
#
#    John Burkardt
#
  import matplotlib.pyplot as plt
  import numpy as np
  import platform

  print ( '' );
  print ( 'code02():' )
  print ( '  Python version: %s' % ( platform.python_version ( ) ) )
  print ( '  Use line plots to display variation in price of three' )
  print ( '  quantities over time.' )
#
#  Read the data.
#
  data = np.genfromtxt ( 'price_data.csv', dtype = None, delimiter=',', names = True )
#
#  Extract price columns for bananas, gas, and milk.
#
  bananas = data['Bananas']
  gas = data['Gas']
  milk = data['Milk']

  n = len ( bananas )
  index = np.linspace ( 1, n, n )
#   
#  Plot the results
#
  plt.plot ( index, bananas, linewidth = 3 )
  plt.plot ( index, gas, linewidth = 3 )
  plt.plot ( index, milk, linewidth = 3 )
  plt.grid ( True )
  plt.legend ( [ 'Bananas', 'Gas', 'Milk' ] )
  plt.xlabel ( '<-- Month index -->', fontsize = 16 )
  plt.ylabel ( '<-- Price ($) -->', fontsize = 16 )
  plt.title ( 'Prices, Feb 2008-Feb 2018', fontsize = 16 )
#
#  Save a copy of the plot.
#
  filename = 'price_plots.png'
  plt.savefig ( filename )
  print ( '  Graphics saved as "%s"' % ( filename ) )
#
#  Display the plot.
#
  plt.show ( block = False )

  plt.clf ( )
#
#  Terminate.
#
  print ( '' )
  print ( 'code02():' )
  print ( '  Normal end of execution.' )

  return

if ( __name__ == '__main__' ):
  code02 ( )