python - 将图例添加到使用 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. 重新定义您的函数以返回其参数的绝对值。