Tikzplotlib无法导出Matplotlib PyPlot颜色栏问题求助

1. 修复等高线draw=none问题 #

tikzplotlib 默认会给 contour 生成的 TikZ 代码添加 draw=none，导致等高线不显示——这是因为它误将线条型等高线识别为填充类元素。解决方法是在调用 contour 时显式指定 linewidths 参数，让 tikzplotlib 正确识别这是需要绘制的线条：

cf = ax.contour(X, Y, Z, cmap=mycmap1, linewidths=1)

2. 导出颜色栏（Colorbar） #

tikzplotlib 对纯线条型 contour 的颜色栏支持存在局限性，因为颜色栏通常与填充区域的数值映射绑定。要保留原 Python 中的颜色栏样式，无需手动编写 TeX 代码，可通过以下方式解决：

方法：结合contourf添加透明填充层 #

添加一个完全透明的填充等高线层，让 tikzplotlib 识别到颜色映射并导出对应的颜色栏，同时保留原线条等高线的样式：

# 添加透明填充层，仅用于触发颜色栏导出
ax.contourf(X, Y, Z, cmap=mycmap1, alpha=0)
# 绘制原线条等高线
cf = ax.contour(X, Y, Z, cmap=mycmap1, linewidths=1)
plt.colorbar(cf, ax=ax)

另外建议先升级 tikzplotlib 到最新版本，旧版本对颜色栏的支持存在明显缺陷：

pip install --upgrade tikzplotlib

完整修正代码 #

import matplotlib.pyplot as plt
import tikzplotlib
import numpy as np

x_min=0
x_max=1
y_min=0
y_max=1
n_x=7
n_y=7
x=np.linspace(x_min,x_max,n_x)
y=np.linspace(y_min,y_max,n_y)
X, Y = np.meshgrid(x, y)

fig, ax = plt.subplots(1,1, figsize=(12,6))

Z=X+Y
mycmap1 = plt.get_cmap()
ax.set_aspect('equal')

# 透明填充层触发颜色栏识别
ax.contourf(X, Y, Z, cmap=mycmap1, alpha=0)
# 显式设置线宽避免draw=none
cf=ax.contour(X,Y,Z,cmap=mycmap1, linewidths=1)
plt.colorbar(cf, ax=ax)

tikzplotlib.save("solution.tex", standalone=True)
plt.show()

导出的 TeX 代码会自动包含颜色栏，且等高线无需手动修改 draw=none 即可正常显示。

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