使用matplotlib contourf绘制电极阵列时自定义非归一化色卡遇错求助

要实现你需要的电极阵列可视化，关键是创建自定义离散颜色映射并配合归一化规则，直接给contourf传入颜色函数是不被支持的，正确实现步骤如下：

1. 导入必要模块 #

import matplotlib.pyplot as plt
from matplotlib.colors import ListedColormap, BoundaryNorm

2. 定义数值-颜色对应关系 #

# 对应数据值：载体(-1)、未激活电极(0)、激活电极(1)
color_list = ["grey", "black", "white"]
value_list = [-1, 0, 1]

3. 创建自定义色卡与归一化规则 #

ListedColormap用于生成离散色卡，BoundaryNorm用于固定数值到颜色的映射区间，避免matplotlib自动缩放颜色范围：

# 创建自定义色卡
cmap = ListedColormap(color_list)
# 设置边界，确保每个数值对应唯一颜色区间
bounds = [-1.5, -0.5, 0.5, 1.5]
norm = BoundaryNorm(bounds, cmap.N)

4. 调用contourf绘图 #

plt.contourf(self.x, self.y, self.activity_meshgrid, cmap=cmap, norm=norm)
# 可选：添加颜色条，标注对应含义
cbar = plt.colorbar(ticks=value_list)
cbar.set_ticklabels(["载体材料", "未激活电极", "激活电极"])
plt.show()

错误原因说明 #

你之前的代码报错是因为：

• contourf的cmap参数要求传入Colormap对象，而非直接返回颜色值的函数
• 当你把数组self.activity_meshgrid传入cmap()函数时，val == 0会返回布尔数组，if判断布尔数组会触发歧义错误（需要用.any()/.all()），但这种方式本身不符合contourf的参数要求

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