python - Matplotlib:对于浮点来说值太小的对数刻度

Question

我想用 matplotlib 绘制一个具有非常小的值的图，以对数刻度显示。直到数字变得太小并且不适合 float 之前，这种方法都可以正常工作。

我使用 SymPy 任意精度 float 表示值，但这些值显然在 matplotlib 内部转换为 Python 或 NumPy 机器 float 。

例如

>>> import sympy
>>> import matplotlib.pyplot as plt
>>> plt.plot([0, 1, 2], [sympy.Float('1e-20'), sympy.Float('1e-100'), sympy.Float('1e-700')])
[<matplotlib.lines.Line2D object at 0x11ac0c208>]
>>> plt.yscale('log')

产生

它应该显示的是 10^-700 处的第三个值(不是负无穷大)。

现在我对让 matplotlib 在内部使用 SymPy Floats 的希望很小(如果可能的话，请告诉我)。我想做的是向 matplotlib 提供该值的对数，我可以自己计算得很好，但仍然以对数刻度在 y 轴上显示该值的指数。

Answer 1

您可以进行对数计算，然后绘制数据，如果您想要相同的刻度标签，您可以使用FuncFormatter:

import sympy
import matplotlib.pyplot as plt
from matplotlib.ticker import FuncFormatter

x = [0, 1, 2]
y = [sympy.Float('1e-20'), sympy.Float('1e-100'), sympy.Float('1e-700')]

def log_formatter(x, pos):
    return "$10^{{{:d}}}$".format(int(x))

formatter = FuncFormatter(log_formatter)

fig, ax = plt.subplots()

y2 = list(map(lambda x:sympy.log(x, 10), y))
ax.plot(x, y2)
ax.yaxis.set_major_formatter(formatter)
ax.grid();

编辑

要添加对数刻度小刻度，您可以创建一个定位器:

import sympy
import matplotlib.pyplot as plt
from matplotlib.ticker import FuncFormatter, Locator

x = [0, 1, 2]
y = [sympy.Float('1e-20'), sympy.Float('1e-100'), sympy.Float('1e-700')]

def log_formatter(x, pos):
    return "$10^{{{:d}}}$".format(int(x))

class LogMinorLocator(Locator):
    def __call__(self):
        majorlocs = self.axis.get_majorticklocs()
        step = majorlocs[1] - majorlocs[0]
        res = majorlocs[:, None] + np.log10(np.linspace(1, 0.1, 10)) * step
        return res.ravel()

formatter = FuncFormatter(log_formatter)

fig, ax = plt.subplots(figsize=(12, 8))
y2 = list(map(lambda x:sympy.log(x, 10), y))
ax.plot(x, y2)
ax.minorticks_on()
ax.yaxis.set_major_formatter(formatter)
ax.yaxis.set_minor_locator(LogMinorLocator())
ax.grid();

这是输出: