python - 将 x 的函数作为 x 刻度标签

Question

这个问题已经有答案了:

What is the correct way to replace matplotlib tick labels with computed values? (1 个回答)

已关闭 9 年前。

假设我有两个 numpy 数组 x和y ，我想绘制一条简单的曲线 y作为 x 的函数。在y轴，我想放置(作为标签) y 的值，但在 x axis 我想将其中值的一些函数作为标签。

例如，如果 x=array([1, 2, 4, 8, 16])和y=array([1, 2, 1, 2, 1]) ，我想为 xticks 分配标签，这将是以下字符串格式的结果:

lambda x_val: "$2^{{+{:.0f}}}$".format(log2(x_val))

但我对通用解决方案感兴趣。

Answer 1

使用matplotlib.ticker.FuncFormatter。无耻地复制和改编custom ticker例如，这样的事情可以工作:

from matplotlib.ticker import FuncFormatter
import matplotlib.pyplot as plt
from matplotlib import rc
import numpy as np

rc('text', usetex=True)

formatter = FuncFormatter(lambda x_val, tick_pos: "$2^{{+{:.0f}}}$".format(np.log2(x_val)))

x = np.array([1, 2, 4, 8, 16])
y = np.array([1, 2, 1, 2, 1])
fig, ax = plt.subplots()
ax.xaxis.set_major_formatter(formatter)
plt.plot(x, y)
plt.show()

结果

注意第一个标签是坏的；运行代码时会发出被零除的警告。这是因为 matplotlib 将轴缩放在 0 到 16 之间，并在 0 处放置刻度线(然后将其传递给格式化程序)。您可以关闭该刻度线，或以不同的方式缩放 x 轴以避免这种情况。