In [1]:
from __future__ import division, print_function
# Load data
import poi_id
%matplotlib inline
import plotting
import matplotlib.pyplot as plt 

enron_data = poi_id.load_dataset()
enron_data.pop("TOTAL")
enron_data.pop("THE TRAVEL AGENCY IN THE PARK")
Out[1]:
{'bonus': 'NaN',
 'deferral_payments': 'NaN',
 'deferred_income': 'NaN',
 'director_fees': 'NaN',
 'email_address': 'NaN',
 'exercised_stock_options': 'NaN',
 'expenses': 'NaN',
 'from_messages': 'NaN',
 'from_poi_to_this_person': 'NaN',
 'from_this_person_to_poi': 'NaN',
 'loan_advances': 'NaN',
 'long_term_incentive': 'NaN',
 'other': 362096,
 'poi': False,
 'restricted_stock': 'NaN',
 'restricted_stock_deferred': 'NaN',
 'salary': 'NaN',
 'shared_receipt_with_poi': 'NaN',
 'to_messages': 'NaN',
 'total_payments': 362096,
 'total_stock_value': 'NaN'}
In [2]:
variables = [v for v in next(enron_data.itervalues()).keys() if v not in ["email_address", "poi"]]

plt.style.use("ggplot")
for v in variables:
    plt.clf()
    plotting.histogram(enron_data, v)
    plt.show()
    plt.clf()
    plotting.boxplot_poi(enron_data, v)
    plt.show()
In [13]:
plotting.boxplot_poi(enron_data, "ratio_stocks_salary")
In [14]:
plotting.boxplot_poi(enron_data, "total_stock_value")