python - Shiny 的 Python 3D 表面图的 Phong 着色

Question

我正在尝试在 Python 中使用镜面阴影创建美观的 3D 图，到目前为止，我已经尝试将 Matplotlib 与 3D 轴和来自 Mayavi 的曲面图一起使用，例如来自 Mayavi 冲浪示例网页:

结果看起来不错，而且在 Mayavi 中似乎确实对照明进行了合理的控制，尽管我似乎无法实现“ Shiny ”的外观。

在 Matlab 中，这可以通过使用“Phong”照明来实现:

参见 http://www.mathworks.com/matlabcentral/fileexchange/35240-matlab-plot-gallery-change-lighting-to-phong/content/html/Lighting_Phong.html

因此，我的问题是:如何在基于 Python 的 3D 绘图中实现这种 Phong 风格的 Shiny 阴影？

Answer 1

如@ben 所述，您可以使用 Mayavi然后交互地改变照明。一个好主意是单击记录脚本按钮，然后您可以在脚本中使用这些代码行(这就是我在这里为 Mayavi 部分所做的)。

另一种选择是使用 Matplotlib .基于shading example ，我设法生成了一个带光照的表面。

请参阅下面的代码。

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

## Test data: Matlab `peaks()`
x, y = np.mgrid[-3:3:150j,-3:3:150j]
z =  3*(1 - x)**2 * np.exp(-x**2 - (y + 1)**2) \
   - 10*(x/5 - x**3 - y**5)*np.exp(-x**2 - y**2) \
   - 1./3*np.exp(-(x + 1)**2 - y**2) 

## Mayavi
surf = mlab.surf(x, y, z, colormap='RdYlBu', warp_scale='auto')
# Change the visualization parameters.
surf.actor.property.interpolation = 'phong'
surf.actor.property.specular = 0.1
surf.actor.property.specular_power = 5



## Matplotlib
fig = plt.figure()
ax = fig.gca(projection='3d')

# Create light source object.
ls = LightSource(azdeg=0, altdeg=65)
# Shade data, creating an rgb array.
rgb = ls.shade(z, plt.cm.RdYlBu)
surf = ax.plot_surface(x, y, z, rstride=1, cstride=1, linewidth=0,
                       antialiased=False, facecolors=rgb)
plt.show()
mlab.show()

结果如下:

• 马亚维:
• Matplotlib: