python - matplotlib twinx 在循环内

Question

我有一个循环，在特定条件下(例如出现 3 次)，我想用 twinx 绘制两个数据集，所以例如最后我在左 y 上有 3 个图，在右 y 上有 3 个图。当我使用通常的 twinx 时，循环的右 y 值不正确。我应该如何修改此示例代码才能使其正确？非常感谢!

import numpy as np
import matplotlib.pyplot as plt

x=np.linspace(0,10,100)
A=5

fig,ax=plt.subplots()

for i in range(0,10):
  A=i**2
  if(i%3==0):
    ax.plot(x,A*np.sin(x),'-ro')
    ax2=ax.twinx()
    ax2.plot(x,A*x,'-bp')
plt.show()

这是输出。

Answer 1

您需要将“ax2”定义移到循环之外(见下文)。然而，这会产生七个图，四个在左边，三个在右边。您需要更新您的“if”条件。

[![import numpy as np
import matplotlib.pyplot as plt

x=np.linspace(0,10,100)
A=5

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

for i in range(0,10):
  A=i**2
  if(i%3==0):
    ax.plot(x,A*np.sin(x),'-ro')
    ax2.plot(x,A*x,'-bp')
plt.show()][1]][1]