More on Python Classes

10,travel
20,travel
13,travel
15,medical
65,apt
70,car
56,medical

''' views for your categorized data
To start,

from views import Views
v=Views() 

# you will be prompted for input. It should be a csv file containing
# amount,category
# to see the figures
# import matplotlib.pyplot as plt and use plt.show() 
'''
import pandas as pd
from pandas import Series, DataFrame

class Views:
    def __init__(self):
        csv_data=input('location of your csv data > ')
        data=pd.read_csv(csv_data,header=None)
        data.columns=['amount','category']
        self.data=data

    def totals(self):
        data=self.data
        totals_list=[]
        for category,group in data.groupby('category'):
            totals_list.append((category,group['amount'].sum()))
        totals_df=DataFrame(totals_list)
        totals_df.columns=['category','total']
        return totals_df
    
    def category_select(self,category):
        data=self.data
        selection=data[data['category']==category]
        return selection
    
    def plot_selection(self,category):
        return self.category_select(category).plot(kind='bar')    

print(__doc__)