block8.python

#creating plot
fig, ax = plt.subplots(1,1,figsize=(9,6))
c_map = {1: 'r', 0: 'b'}
ax.scatter(enron_df['salary'],enron_df['long_term_incentive'], 
           c=[c_map[i] for i in enron_df['poi']])
plt.ylabel("Long Term Incentive ($)")
plt.xlabel("Salary ($)")
plt.title("Salary vs. Long Term Incentive")

not_poi = mlines.Line2D([], [], color='blue', marker='o',
                          markersize=10, label='Not POI')
poi = mlines.Line2D([], [], color='red', marker='o',
                          markersize=10, label='POI')
plt.legend(handles=[not_poi,poi])

plt.show()