如何在matplotlib中检查轴是否为极坐标投影?

时间:2019-02-19 04:33:37

标签: python matplotlib cartesian

是否有figax中的属性可以指示斧头的投影是否为极性?

我正在尝试在其中一个具有以下功能的更复杂的函数中创建一个基本的嵌套函数:

is_polar(ax):
    return ax.some_attribute

尽管如此,我不确定b / c是否可行,因为我一直在仔细研究他的明显属性。以为我会在进行详尽的手动搜索之前先接触社区。

# Source | https://matplotlib.org/gallery/pie_and_polar_charts/polar_scatter.html
# Fixing random state for reproducibility
np.random.seed(19680801)

# Compute areas and colors
N = 150
r = 2 * np.random.rand(N)
theta = 2 * np.pi * np.random.rand(N)
area = 200 * r**2
colors = theta

fig = plt.figure()
ax = fig.add_subplot(111, projection='polar')
c = ax.scatter(theta, r, c=colors, s=area, cmap='hsv', alpha=0.75)

enter image description here

1 个答案:

答案 0 :(得分:3)

您可以根据matplotlib axes docs使用ax.name检查投影。它以字符串形式包含投影,因此您只需

if ax.name == 'polar':
    # ....