# MATPLOTLIB
import matplotlib.pyplot as plt
import matplotlib
import matplotlib.cm as cm
import matplotlib.patches as patches
N = len(data['species'].unique())
cmap = cm.jet(pd.np.linspace(0, 1, num=N))
colors = {k: cmap[i] for i, k in enumerate(data['species'].unique())}
mmap = ['o', 's', '^']
markers = {k: mmap[i] for i, k in enumerate(data['species'].unique())}
data['color'] = data['species'].apply(lambda x: colors[x])
data['marker'] = data['species'].apply(lambda x: markers[x])
fig, ax = plt.subplots(1, 1, figsize=(7.5, 5))
artists = list()
legends = list()
for k in data['species'].unique():
s = data.loc[data['species']==k]
a = ax.scatter(s['petal_length'], s['petal_width'], marker=markers[k], edgecolor=colors[k], facecolor=colors[k])
legends.append(k)
artists.append(a)
ax.set_xlabel('petal_length')
ax.set_ylabel('petal_width')
ax.legend(artists, legends, loc='upper left')
plt.savefig('./../static/img/iris-matplotlib.png')