Matplotlib：具有从主轴映射的值的辅助轴

Question

我有一个显示x4 vs y的图表：

y是一些其他变量的对数，比如q（即y = log（q））q的值是外行人在阅读该图时将理解的内容。

我想在图的右侧设置一个辅助轴，其中的线与左侧轴处于相同的垂直位置，但标签被等效的q值 - 即exp（y） - 替换为2位小数。

是否有一种有效的方法来创建从主y轴进行此映射的辅助轴？

我正在使用来自pyplot的matplotlib。一些示例启动代码：

import numpy as np
import matplotlib.pyplot as plt

q = np.random.rand(100,)
y = np.log(q)
x4 = np.random.rand(100,)

plt.scatter(x4, y)

我想添加第二个轴，它具有与主轴相同的垂直位置和间距，但标签被exp（y）替换。还包括轴标签“q”。

谢谢

Answer 1

我的建议是使用双轴并与原始轴共用以固定刻度位置。然后，您可以使用FuncFormatter为刻度线指定正确的标签。这样做的好处是您不需要先验地修复绘图的限制，并且可以在绘图内自由缩放和平移。

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


f = lambda q: np.log(q)
finv = lambda x: np.exp(x)

x = np.random.rand(100,)
y = f(np.random.rand(100,))


fig, ax = plt.subplots()
ax2 = ax.twinx()
ax.get_shared_y_axes().join(ax,ax2)

ax.scatter(x, y)

ax2.yaxis.set_major_formatter(FuncFormatter(lambda x,pos: f"{finv(x):.2f}"))
plt.show()

Answer 2

您可以使用双轴。这个想法如下：

• 创建双轴并在轴上绘制相同的数据。这将确保在两个y轴上的刻度位置都是相同的。
• 获取作为字符串的刻度标签，将它们转换为整数并使用np.exp()将它们的指数四舍五入到最多2位小数。
• 将新标签分配到右侧y轴
• 您还需要确保两个轴具有相同数量的刻度和相同的范围，如下面的注释中@ImportanceOfBeingEarnest所建议的那样

---

import numpy as np; np.random.seed(123)
import matplotlib.pyplot as plt

fig, ax = plt.subplots()

q = np.random.rand(100,)
y = np.log(q)
x4 = np.random.rand(100,)

ax.scatter(x4, y)
ax.set_xlabel('x4', fontsize=16)
ax.set_ylabel('y', fontsize=16)

ax2 = ax.twinx()
ax2.scatter(x4, y)

ax2.set_yticks(ax.get_yticks())
ax2.set_ylim(ax.get_ylim())
fig.canvas.draw()

labels = [int(i.get_text()[1:-1]) for i in ax2.get_yticklabels()]
labels_new = [np.round(np.exp(i), 2) for i in labels]

ax2.set_yticklabels(labels_new);
ax2.set_ylabel('q', fontsize=16)
plt.show()