matplotlib 3d添加光源

要在matplotlib 3D图形中添加光源，需要使用`mplot3d.art3d.LightSource`类。该类提供了一种将光源应用于3D图形的方法。以下是添加光源的示例代码：

import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
import numpy as np
from matplotlib.colors import LightSource

# 创建3D图形
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

# 生成一些随机数据
x, y = np.linspace(-5, 5, 100), np.linspace(-5, 5, 100)
X, Y = np.meshgrid(x, y)
Z = np.sin(np.sqrt(X**2 + Y**2))

# 创建LightSource对象
ls = LightSource(azdeg=0, altdeg=65)

# 计算阴影
rgb = ls.shade(Z, cmap=plt.cm.gist_earth, vert_exag=0.1, blend_mode='soft')

# 绘制3D图形
surf = ax.plot_surface(X, Y, Z, rstride=1, cstride=1, facecolors=rgb, linewidth=0, antialiased=False)

plt.show()

在这个例子中，我们创建了一个3D图形，并使用`LightSource`对象创建了光源。`shade()`方法用于计算阴影，并将结果作为RGB颜色数组传递给`plot_surface()`方法。最终结果是一个具有光照效果的3D图形。