如何将Matplotlib图形转换为极坐标图？环形扇区绘制异常问题

我尝试在Matplotlib中制作环形扇区，最终目的是掌握如何将笛卡尔坐标系下的复杂图形转换为环形扇区。以下是我尝试的代码：

import numpy as np
theta1=45
theta2=100

# Radius coordinates

r1 = 0.4
r2 = 0.5

# Plot using bar()
height = r2 - r1
theta_mid = np.deg2rad((theta1 + theta2)/2)
theta_width = np.deg2rad(theta2 - theta1)
fig, ax = plt.subplots(figsize=[6,6], subplot_kw={'projection': 'polar'})
p = PatchCollection([Rectangle((np.deg2rad(theta1), r1), theta_width, height, color='blue')])
ax.add_collection(p)

但目前仅在极坐标图上显示出一个四边形，边缘仍为直线。请问我哪里操作出错了？

解决方法 #

问题根源 #

你使用的Rectangle是笛卡尔坐标系专属的图形补丁，Matplotlib不会自动将其转换为极坐标下的环形扇区。把笛卡尔矩形直接添加到极坐标轴中，它只会保留原有的直线边缘，不会适配极坐标的曲线特性。

方案1：直接绘制环形扇区（推荐） #

如果只是需要环形扇区，直接使用Matplotlib提供的Wedge补丁即可——它原生支持极坐标下的扇形/环形扇区绘制：

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.patches import Wedge

theta1 = 45
theta2 = 100
r1 = 0.4
r2 = 0.5

fig, ax = plt.subplots(figsize=[6,6], subplot_kw={'projection': 'polar'})
# Wedge参数：(中心坐标, 外半径, 内半径, 起始角度(度), 结束角度(度))
wedge = Wedge((0, 0), r2, r1, theta1, theta2, color='blue')
ax.add_patch(wedge)

# 调整轴范围确保图形完整显示
ax.set_ylim(0, r2 + 0.1)
plt.show()

方案2：将笛卡尔复杂图形映射为环形扇区 #

如果你的最终目标是把任意笛卡尔图形转换为环形扇区（极坐标映射），需要手动完成坐标转换，步骤如下：

1. 提取笛卡尔图形的所有顶点坐标
2. 将每个顶点的笛卡尔坐标(x,y)转换为极坐标(theta, r)：
   • theta = np.arctan2(y, x)（弧度制）
   • r = np.sqrt(x² + y²)
3. 将原笛卡尔图形的r范围映射到目标环形的[r1, r2]区间
4. 使用转换后的极坐标顶点创建路径补丁

以下是将笛卡尔矩形转换为环形扇区的示例代码：

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.path import Path
from matplotlib.patches import PathPatch

# 定义笛卡尔坐标系下的矩形顶点
cartesian_verts = np.array([[0,0], [1,0], [1,1], [0,1], [0,0]])
# 转换为极坐标
theta = np.arctan2(cartesian_verts[:,1], cartesian_verts[:,0])
r = np.sqrt(cartesian_verts[:,0]**2 + cartesian_verts[:,1]**2)
# 将原r范围映射到[r1, r2]
r_mapped = r1 + (r2 - r1) * (r / np.max(r))

# 组合成极坐标下的顶点数组（极坐标轴的坐标格式为(theta, r)）
polar_verts = np.column_stack((theta, r_mapped))

# 创建路径和补丁
path = Path(polar_verts)
patch = PathPatch(path, color='blue', alpha=0.5)

fig, ax = plt.subplots(figsize=[6,6], subplot_kw={'projection': 'polar'})
ax.add_patch(patch)
ax.set_ylim(0, r2 + 0.1)
plt.show()

内容的提问来源于stack exchange，提问作者Vasilis Lemonidis