import numpy as np
import operator
import matplotlib.pyplot as plt
def createDateSet():
group = np.array([[1.0,1.1],[1.0,1.0],[0,0],[0,0.1]])
labels = ['A','A','B','B']
return group,labels
group,labels = createDateSet()
print(group)
print(labels)
c=np.random.randint(0,10,len(group))
plt.figure(figsize=(8,5))
#plt.plot(group[:,0],group[:,1],'ro')
#图一
#plt.scatter(group[:,0],group[:,1],marker='o')
#图二
#plt.scatter(group[:,0],group[:,1],marker='o',c='r')
#图三
#plt.scatter(group[:,0],group[:,1],c=c,marker='o')
#plt.colorbar()
#图四
plt.grid(True)
plt.xlabel('x')
plt.ylabel('y')
plt.title('group')
plt.show()
https://jingyan.baidu.com/article/0aa223757124e088cd0d6444.html