用matplotlib绘制3D图形

-- TOC --

下面是一个简单的3D图形实例:

import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(-5, 5, 10000)
fig = plt.figure('3D graph')
ax = fig.add_subplot(111, projection='3d') 

ax.set_title(r'lines')
ax.plot(x, x, x, linewidth=1, label=r'$z=y=x$')
ax.plot(x, x*2, x*4, linewidth=1, label=r'$z=2y=4x$')
ax.plot(x, np.sin(x), np.cos(x), linewidth=1, label=r'$y=sin(x),z=cos(x)$')
ax.set_xlim(-5, 5)
ax.set_ylim(-5, 5)
ax.set_zlim(-5, 5)
ax.legend()
ax.set_xlabel('x')
ax.set_ylabel('y')
ax.set_zlabel('z')
ax.text(0, 0, 0, '0')  # (0,0,0)

#该函数接受两个参数,第一个参数是竖直旋转,第二个参数是水平旋转,旋转单位是度°
ax.view_init(30, -60)

plt.show()

效果图:

simple_3d

可以用鼠标拖动的方式,任意旋转这个3D图形。

本文链接:https://cs.pynote.net/sf/python/matplotlib/202310025/

-- EOF --

-- MORE --