如何以编程方式查找 Mayavi 颜色图的有效名称？答案

【问题标题】：How can I programmatically find valid names for Mayavi colormaps?如何以编程方式查找 Mayavi 颜色图的有效名称？
【发布时间】：2013-10-30 23:18:02
【问题描述】：

我有一个基于 Mayavi 构建的应用程序，我让用户在其中为某些数据指定颜色图。因为较新版本的 Matplotlib 具有 Mayavi 缺少的颜色图，所以如果 Mayavi 不提供，我希望能够使用本机 Mayavi 颜色图或基于 Matplotlib 颜色图制作查找表。

我可以尝试使用用户提供的名称并捕获引发的 TraitError，但我宁愿仅根据现有颜色列表检查提供的名称。例如，在 Matplotlib 中我会这样做

import matplotlib.pyplot as plt

def is_mpl_cmap(name):
    if name in dir(plt.cm):
        return True
    return False

我可以访问 Mayavi 中是否有类似的颜色图名称存储库？

【问题讨论】：

【解决方案1】：

from mayavi.core import lut_manager

def is_tvtk_cmap(name): return name in lut_manager.lut_mode_list()

【讨论】：