%matplotlib inline import math import copy import numpy as np import pandas as pd from matplotlib import pylab as plt df = pd.DataFrame( { 'x1': [1.5, 2, 3, 1.5, 0.5, -1, -2, -3, -1.5, 0], 'x2': [1, 2.5, 3, -2, 2, -3, -1.2, -0.5, 2, -1.5], 'label': ['A', 'A', 'A', 'A', 'A', 'B', 'B', 'B', 'B', 'B'], 'label_index': [1.0, 1.0, 1.0, 1.0, 1.0, -1.0, -1.0, -1.0, -1.0, -1.0] } ) df label_a = df[df['labe