python数据分析 | Matplotlib3D绘图

1 导入模块

from matplotlib import pyplot as plt
import numpy as np
from mpl_toolkits.mplot3d import Axes3D
%matplotlib notebook # 在notebook展示和旋转图片

2 3D曲线图

# 将 0-15 分为 1000份
zline = np.linspace(0,15,1000)
xline = np.sin(zline)
yline = np.cos(zline)

# 创建3D画布
fig = plt.figure()
ax = Axes3D(fig)

# 绘制图形
ax.plot(xline,yline,zline)
plt.show()

3 散点图

x = np.random.rand(100)
y = np.random.rand(100)
z = np.random.rand(100)

fig = plt.figure()
ax = Axes3D(fig)
ax.scatter(x,y,z,color="b")
plt.show()

4 平面图

# 构建数据
x = [1,2,3,4]
y = [1,2,3,4]

X,Y = np.meshgrid(x,y)  # x --> 行的延申（复制）， y --> 列的延申（复制）
print(X)
print(Y)
"""
结果：
[[1 2 3 4]
 [1 2 3 4]
 [1 2 3 4]
 [1 2 3 4]]
[[1 1 1 1]
 [2 2 2 2]
 [3 3 3 3]
 [4 4 4 4]]
"""

fig = plt.figure()
ax = Axes3D(fig)

# 绘制平面  z=3
ax.plot_surface(X,Y,Z=X*0+4) # 平面放置在z=4的位置上 平面的函数时z = X*0 + 4

# 设置x,y,z标签
ax.set_xlabel("x")
ax.set_ylabel("y")
ax.set_zlabel("z")

plt.show()

5 曲面图

# 构建数据
x = np.arange(-2,2,0.1)
y = np.arange(-2,2,0.1)
print(x)
print(y)

# 延申，构造平面
X,Y = np.meshgrid(x,y)
print(X)
print(Y)

# 定义构造曲面的函数，即z
def fun(x,y):
#     return x**2
    return x**2 + y**2

fig = plt.figure()
ax = Axes3D(fig)

# 绘制曲面 
ax.plot_surface(X,Y,fun(X,Y),rstride=1,cstride=1) # 平面放置在z=4的位置上
plt.show()

这个图，本人觉得很好看。
学习来自使用matplotlib绘制3D函数图像

import matplotlib.pyplot as plt  # 绘图用的模块
from mpl_toolkits.mplot3d import Axes3D  # 绘制3D坐标的函数
import numpy as np


def fun(x, y):
    return np.power(x, 2) + np.power(y, 2)


fig1 = plt.figure()  # 创建一个绘图对象
ax = Axes3D(fig1)  # 用这个绘图对象创建一个Axes对象(有3D坐标)
X, Y = np.mgrid[-2:2:40j, -2:2:40j]  # 从-2到2分别生成40个取样坐标，并作满射联合
Z = fun(X, Y)  # 用取样点横纵坐标去求取样点Z坐标
plt.title("This is main title")  # 总标题
ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap=plt.cm.coolwarm, alpha=0.5)  # 用取样点(x,y,z)去构建曲面
ax.set_xlabel('x label', color='r')
ax.set_ylabel('y label', color='g')
ax.set_zlabel('z label', color='b')  # 给三个坐标轴注明
plt.show()  # 显示模块中的所有绘图对象