如何在Matplotlib中使用Plotly的自定义色阶（如ice、dense）？

问题核心是Plotly返回的色阶是rgb(x,y,z)格式的字符串，而Matplotlib的ListedColormap需要的是归一化到0-1的浮点RGB数组或0-255的整数RGB数组，直接传入字符串会触发ValueError。

下面提供两种可靠的解决方法：

方法一：使用Plotly内置转换工具（推荐） #

Plotly提供了convert_colors_to_same_type函数，可以直接将色阶字符串转换为Matplotlib兼容的格式：

import plotly.express as px
import matplotlib.pyplot as plt
import matplotlib.colors as mcolors

# 获取Plotly的ice色阶并转换为浮点型RGB数组（0-1范围）
ice_colors = px.colors.convert_colors_to_same_type(
    px.colors.sequential.ice,
    colortype='float'
)[0]

# 创建Matplotlib可用的自定义色阶
my_cmap = mcolors.ListedColormap(ice_colors, name='ice')

# 测试绘图
plt.imshow([[1,2,3],[4,5,6],[7,8,9]], cmap=my_cmap)
plt.colorbar()
plt.show()

方法二：手动解析RGB字符串 #

如果需要自定义处理逻辑，可以通过正则表达式提取RGB数值并归一化：

import plotly.express as px
import matplotlib.pyplot as plt
import matplotlib.colors as mcolors
import re

# 解析单个rgb字符串为0-1范围的浮点数组
def parse_rgb_string(rgb_str):
    # 提取括号内的数字部分
    rgb_nums = re.findall(r'\d+', rgb_str)
    # 转换为整数后归一化到0-1
    return [int(num) / 255 for num in rgb_nums]

# 批量转换整个色阶列表
ice_colors = [parse_rgb_string(color) for color in px.colors.sequential.ice]

# 创建自定义色阶
my_cmap = mcolors.ListedColormap(ice_colors, name='ice')

# 测试绘图
plt.imshow([[1,2,3],[4,5,6],[7,8,9]], cmap=my_cmap)
plt.colorbar()
plt.show()

两种方法都能成功将Plotly的ice色阶导入Matplotlib，方法一更简洁且不易出错，优先推荐使用。

内容的提问来源于stack exchange，提问作者JD.