cusCohort9.py

# Average spending price per period
df['TotalSale'] = df['Price'] * df['Quantity'] #Calculation total sale
dfPrice = df.groupby(
    ['cohortMonth', 'cohortIndex'],
    as_index = False
)['TotalSale'].mean()

dfPrice = pd.pivot_table(
    dfPrice,
    values = 'TotalSale',
    index = ['cohortMonth'],
    columns = ['cohortIndex']
) # Pivot

# Plot
plotTable(dfPrice, 'Total spending by month', percent = False)