將圖例添加到使用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. 重新定義您的函數以返回其參數的絕對值。