建立自己的数据集以及使用matplotlib画图

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