[Python] Read and plot data from csv file

Install:

pip install pandas
pip install matplotlib # check out the doc from site

import pandas as pd
import matplotlib.pyplot as plt
from numpy import mean
 
def load_df(symbol):
    return pd.read_csv("data/{0}.csv".format(symbol))
 
def get_max_close(symbol):
    """ Return the maximum closing value for stock idicicated by symbol.
    Note: Data for a stock is stored in file: data/<symbol>.csv
    """
    df = load_df(symbol)
    return df["Close"].max()
 
def get_mean_volume(symbol):
    df = load_df(symbol)
    return mean(df['Volume'])
 
def printAdjClose():
    df = pd.read_csv('data/ibm.csv')
    df[['Low', 'High']].plot()
    plt.show()
 
def test_run():
    """
    Function called by Test Run
    """
    for symbol in ['aapl', 'ibm']:
        print("Max close")
        print(symbol, get_max_close(symbol))
        print("Mean Volume")
        print(symbol, get_mean_volume(symbol))
 
if __name__ == "__main__":
    test_run()
    printAdjClose()