将图例添加到使用mpmath制作的复杂图表中

Question

我一直在使用mpmath库中的cplot命令绘制复杂函数。它的工作非常简单，我所要做的就是写cplot(G_fit, [0.001, v_max], [-v_max, v_max], points = 100000)，我得到一个相当平滑的函数G_fit超过0.001 < Re(z) < v_max和-v_max < Im(y) < v_max的图表。但有两个问题：

1. 如何添加图例？它在文档中说，功能的大小由亮度表示，相位角是色调。那很好，但是的大小和色调？我想要做的就是在图表旁边添加一个从黑色到白色的条形图，并给出这些亮度的数值。

2. 有没有办法只绘制幅度（即亮度？）

谢谢，对不起，如果这是微不足道的，但我在文档中找不到任何相关内容。

Answer 1

我自己的研究发现：

1. 看起来不像。

2. 重新定义函数以返回其参数的绝对值。