python——类别数据,根据类别自动选择颜色

1147-柳同学

发表文章数:589

首页 » 算法 » 正文

import numpy as np
import matplotlib as mpl
import matplotlib.pyplot as plt


# 画决策边界
def plotDescionBoundary(X, y, theta):
    # 样本点颜色
    cm_dark = mpl.colors.ListedColormap(['g', 'r'])
    plt.xlabel('Exam 1 score')
    plt.ylabel('Exam 2 score')
    # 根据y的结果自动的在cmap中选择颜色,c参数代表颜色
    plt.scatter(X[:, 0], X[:, 1], c=np.array(y).squeeze(), cmap=cm_dark, s=30)

    # 画分类决策面
    x1 = np.arange(min(X[:, 0]), max(X[:, 0]), 0.1)
    x2 = -(theta[0] + theta[1] * x1) / theta[2]
    plt.plot(x1, x2)
    plt.show()


if __name__ == '__main__':
    X = np.array([[34.62365962451697, 78.0246928153624],
                  [30.28671076822607, 43.89499752400101],
                  [60.18259938620976, 86.30855209546826],
                  [79.0327360507101, 75.3443764369103]])
    y = [0, 0, 1, 1]
    theta = [[-247.44837176],
             [2.78836624],
             [1.56814101]]
    plotDescionBoundary(X, y, theta)

python——类别数据,根据类别自动选择颜色

未经允许不得转载:作者:1147-柳同学, 转载或复制请以 超链接形式 并注明出处 拜师资源博客
原文地址:《python——类别数据,根据类别自动选择颜色》 发布于2021-01-07

分享到:
赞(0) 打赏

评论 抢沙发

评论前必须登录!

  注册