Mayavi：在三角网格中插值面颜色

Question

我拼凑了以下代码，用来绘制一个三角形网格，并且用一个额外的标量函数来指定颜色：

#! /usr/bin/env python
import numpy as np
from mayavi import mlab

# Create cone
n = 8
t = np.linspace(-np.pi, np.pi, n)
z = np.exp(1j*t)
x = z.real.copy()
y = z.imag.copy()
z = np.zeros_like(x)
triangles = [(0, i, i+1) for i in range(n)]
x = np.r_[0, x]
y = np.r_[0, y]
z = np.r_[1, z]
t = np.r_[0, t]

# These are the scalar values for each triangle
f = np.mean(t[np.array(triangles)], axis=1)

# Plot it
mesh = mlab.triangular_mesh(x, y, z, triangles,
                            representation='wireframe',
                            opacity=0)
cell_data = mesh.mlab_source.dataset.cell_data
cell_data.scalars = f
cell_data.scalars.name = 'Cell data'
cell_data.update()

mesh2 = mlab.pipeline.set_active_attribute(mesh,
        cell_scalars='Cell data')
mlab.pipeline.surface(mesh2)

mlab.show()

这个方法效果还不错。不过，我希望每个三角形不是都有统一的颜色，并且三角形之间的过渡太生硬了。我更希望整个表面颜色能够平滑过渡。

有没有办法做到这一点呢？

Answer 1

我觉得你应该使用点数据，而不是单元格数据。单元格数据中的单个数值并不对应到某个具体的点，而是分配给整个面。看起来你只是想把t数据分配给顶点。默认情况下，点的数值会在每个面之间平滑地过渡。

point_data = mesh.mlab_source.dataset.point_data
point_data.scalars = t
point_data.scalars.name = 'Point data'
point_data.update()

mesh2 = mlab.pipeline.set_active_attribute(mesh,
        point_scalars='Point data')