python 三维散点图拟合曲面_[Python] 三维散点/曲线/曲面 ( 含有笔记、代码、注释 )...

3D散点

import matplotlib as mpl

import matplotlib.pyplot as plt

from mpl_toolkits.mplot3d import Axes3D

import numpy as np

xs1 = np.random.randint(30,40,100)

ys1 = np.random.randint(20,30,100)

zs1 = np.random.randint(10,20,100)

xs2 = np.random.randint(50,60,100)

ys2 = np.random.randint(30,40,100)

zs2 = np.random.randint(50,70,100)

xs3 = np.random.randint(10,30,100)

ys3 = np.random.randint(40,50,100)

zs3 = np.random.randint(40,50,100)

# 方式1：设置三维图形模式

fig = plt.figure() # 创建一个画布figure，然后在这个画布上加各种元素。

ax = Axes3D(fig) # 将画布作用于 Axes3D 对象上。

ax.scatter(xs1,ys1,zs1) # 画出(xs1,ys1,zs1)的散点图。

ax.scatter(xs2,ys2,zs2,c='r',marker='^')

ax.scatter(xs3,ys3,zs3,c='g',marker='*')

ax.set_xlabel('X label') # 画出坐标轴

ax.set_ylabel('Y label')

ax.set_zlabel('Z label')

plt.show()

3D曲线

import matplotlib as mpl

from mpl_toolkits.mplot3d import Axes3D

import numpy as np

import matplotlib.pyplot as plt

# 设置图例字号

mpl.rcParams['legend.fontsize'] = 10

# 方式2：设置三维图形模式

fig = plt.figure()

ax = fig.gca(projection='3d')

# 测试数据

theta = np.linspace(-4 * np.pi, 4 * np.pi, 100)

z = np.linspace(-4, 4, 100) / 4

r = z**3 + 1

x = r * np.sin(theta)

y = r * np.cos(theta)

# 绘制图形

ax.plot(x, y, z, label='parametric curve')

# 显示图例

ax.legend()

# 显示图形

plt.show()

3D曲线拟合(含噪音)

# 不含噪声散点图

import matplotlib as mpl