python3利用Axes3D库画3D模型图

python3利用利用Axes3D库画库画3D模型图模型图

主要为大家详细介绍了python3利用Axes3D库画3D模型图，文中示例代码介绍的非常详细，具有一定的参考价

值，感兴趣的小伙伴们可以参考一下

Python3利用Axes3D库画3D模型图，供大家参考，具体内容如下

最近在学习机器学习相关的算法，用python实现。自己实现两个特征的线性回归，用Axes3D库进行建模。

python代码

import numpy as np

from scipy import stats

X1 = np.arange(-4, 4, 0.1)

m = len(X1)

X2 = np.random.rand(m)*5

# print(X2)

# print(X1)

# 堆叠全1数组和X1以及X2形成样本的矩阵，倒置，用以矩阵乘法

X = np.vstack((np.full(m, 1), X1, X2)).T

# y = 15*X1 + 3 * X2 + theta0

# 自定义样本输出

Y = X1 + 3 * X2 + 3*np.random.randn(m)

# zip函数构建一个多元祖[(x1,y1),(x2,y2)...],ravel函数将一个多维数组合并成一维数组

Z = np.array([theta[1] * d + theta[2]*p + theta[0] for d, p in zip(np.ravel(M), np.ravel(N))]).reshape(M.shape)

# 根据网格和函数构建图形 suface是一个完整的面

ax.plot_surface(M, N, Z)

# scatter是散点图

ax.scatter(X1, X2, Y, c='r')

# 设置坐标轴的名称

ax.set_xlabel('X')

ax.set_ylabel('Y')

ax.set_zlabel('Z')