使用mpl_toolkits.axes_grid1库中的host_subplot()函数在python中绘制双y轴极坐标图

mpl_toolkits.axes_grid1库是matplotlib库的一个扩展库，用于创建特殊的轴布局和子图网格。其中的host_subplot()函数可以用于创建带有双y轴的极坐标图。

首先，我们需要导入需要的模块：

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.axes_grid1 import host_subplot
import mpl_toolkits.axisartist as AA

接下来，我们可以生成一些随机数据作为例子：

# 生成随机数据
theta = np.linspace(0, 2*np.pi, 100)
r1 = np.random.rand(100)
r2 = np.random.rand(100)

# 将极坐标转换为直角坐标
x1 = r1 * np.cos(theta)
y1 = r1 * np.sin(theta)
x2 = r2 * np.cos(theta)
y2 = r2 * np.sin(theta)

然后，我们可以使用host_subplot()函数创建一个极坐标子图，并设置其布局和轴：

# 创建极坐标子图
ax = host_subplot(111, polar=True)

# 设置右边的y轴属性
ax2 = ax.twin()
ax2.axis["left"].set_visible(False)
ax2.axis["right"].set_visible(True)
ax2.axis["right"].major_ticklabels.set_visible(True)
ax2.axis["right"].label.set_visible(True)

接下来，我们可以在极坐标子图上画出两条曲线，并设置其颜色和样式：

# 绘制曲线
ax.plot(theta, r1, color='r', label='Line1')
ax2.plot(theta, r2, color='b', label='Line2')

# 设置图例
ax.legend(loc='upper right')
ax2.legend(loc='upper left')

最后，我们可以设置一些图形的属性，比如标题和标签：

# 设置标题
ax.set_title("Double Y-Axis Polar Plot")

# 设置标签
ax.set_xlabel("Theta")
ax.set_ylabel("Line 1")

ax2.set_ylabel("Line 2")

最后，我们可以显示图形：

plt.show()

这样，我们就完成了一个双y轴的极坐标图的绘制。完整的代码如下：

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.axes_grid1 import host_subplot
import mpl_toolkits.axisartist as AA

# 生成随机数据
theta = np.linspace(0, 2*np.pi, 100)
r1 = np.random.rand(100)
r2 = np.random.rand(100)

# 将极坐标转换为直角坐标
x1 = r1 * np.cos(theta)
y1 = r1 * np.sin(theta)
x2 = r2 * np.cos(theta)
y2 = r2 * np.sin(theta)

# 创建极坐标子图
ax = host_subplot(111, polar=True)

# 设置右边的y轴属性
ax2 = ax.twin()
ax2.axis["left"].set_visible(False)
ax2.axis["right"].set_visible(True)
ax2.axis["right"].major_ticklabels.set_visible(True)
ax2.axis["right"].label.set_visible(True)

# 绘制曲线
ax.plot(theta, r1, color='r', label='Line1')
ax2.plot(theta, r2, color='b', label='Line2')

# 设置图例
ax.legend(loc='upper right')
ax2.legend(loc='upper left')

# 设置标题
ax.set_title("Double Y-Axis Polar Plot")

# 设置标签
ax.set_xlabel("Theta")
ax.set_ylabel("Line 1")

ax2.set_ylabel("Line 2")

# 显示图形
plt.show()

运行上述代码后，我们就可以看到一个带有双y轴的极坐标图。其中，红色曲线代表Line1，蓝色曲线代表Line2，两个曲线分别对应左右两个y轴。