import numpy as np
import matplotlib.pyplot as plt
data = np.array( [ [ 6 , 7],[ 2 , 3],[ 3 , 7],[ 4 , 4],[ 5 , 8],[ 6 , 5],[ 7 , 9],[ 8 , 5],[ 8 , 2],
[10 , 2] ])
categories = np.array([0,1,1,1,1,2,2,2,2,2])
print("categories:", categories)
colormap = np.array(['r', 'g', 'b'])
plt.scatter(data[:,0], data[:,1], s=100, c=colormap[categories])
# s is the size of the dots in 1/72 inches
print(data)
print( "data[:,0]="  , data[:,0])
print( "data[:,1]" , data[:,1])
plt.show()