Matplotlib：如何在现有颜色条上添加标签？

Question

我正在使用一个名为kwant的Python模块，它会自动为我创建一个完整的图，包括一个颜色条。但是，我没有找到任何方法为此颜色栏添加标签。由于我无法访问直接构建绘图的数据，因此我只能希望能够从已创建的图形对象中访问colorbar对象。有可能这样做吗？

Answer 1

https://kwant-project.org/doc/1/reference/generated/kwant.plotter.plot

colorbar : bool
Whether to show a colorbar if colormap is used. Ignored if ax is provided.

https://kwant-project.org/doc/1.0/reference/kwant.plotter

你看过这个吗？

Answer 2

我没有这个模块的经验，但...... 从docs中，绘图函数将来自matplotlib的ax作为输入。 所以你可以做类似

的事情

import matplotlib.pyplot as plt
fig, ax = plt.subplots(1) 
...
out = kwant.some_plot(..., ax = ax)

现在你可以使用matplotlib通过fig和ax访问绘图的不同部分。