python绘制散点图的例子(用于分类结果的区分)

import matplotlib.pyplot as plt

# 分类结果和真实标签值
classifierResult = [0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 0, 1, 1, 2, 1, 1, 3, 1, 1, 2, 1, 1, 0, 2, 2, 2, 1, 0, 3]
classNumber = [0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 2, 2, 2, 2, 2, 2, 3, 3, 3, 3]
plt.figure(figsize=(8, 4))  # 创建绘图对象
classifierResult1 = []
classNumber1 = []
classifierResult2 = []
classNumber2 = []
for i in range(len(classifierResult)):
    if classifierResult[i] == classNumber[i]:
        classifierResult1.append(classifierResult[i])
        classNumber1.append(classNumber[i])

    else:

        classifierResult2.append(classifierResult[i])
        classNumber2.append(classNumber[i])

plt.scatter(classifierResult1, classNumber1, c='g')  # 在当前绘图对象绘图(X轴,Y轴,蓝色虚线,线宽度)
plt.scatter(classifierResult2, classNumber2, c='r')
plt.title("scatter", fontsize=22)  # 标题
plt.xlabel("classifierResult", fontsize=12)  # X轴标签
plt.ylabel("classNumber", fontsize=12)  # Y轴标签
plt.show()

你可能感兴趣的:(python绘制散点图的例子(用于分类结果的区分))