Добавление легенды в сложный сюжет, созданный с помощью mpmath

Question

Я использовал команду cplot из библиотеки mpmath для построения сложной функции. Это работает довольно легко, все, что мне нужно сделать, это написать 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. Переопределите вашу функцию, чтобы вернуть абсолютное значение ее аргумента.