Python中RGB极坐标图绘制问题：imshow输出空白如何解决？

你遇到的问题根源在于imshow是为笛卡尔（直角）坐标系设计的工具，直接在极坐标子图中用它并设置extent参数，并不会自动完成坐标系统的转换，所以才会出现空白的圆形输出。下面给你两种可行的实现方案：

方法一：极坐标网格+pcolormesh映射RGB颜色 #

这种方案适合你的数据按「半径方向×角度方向×3通道」组织的场景，我们先生成极坐标网格，再把RGB数据直接映射到网格点上：

import matplotlib.pyplot as plt
import numpy as np

# 示例RGB数据：N（半径分层）×M（角度分段）×3
data = np.array([[[0,0,1],[0,1,0],[1,0,0]],
                 [[0,0,0.5],[0,0.5,0],[0.5,0,0]]])

# 生成极坐标网格的基础参数
n_radii = data.shape[0]
n_angles = data.shape[1]
radii = np.linspace(0, 1, n_radii)
angles = np.linspace(0, 2*np.pi, n_angles, endpoint=False)

# 构建极坐标网格（角度×半径）
angles_grid, radii_grid = np.meshgrid(angles, radii)

# 将RGB数据重塑为每个网格点对应的颜色数组
colors = data.reshape(-1, 3)

# 创建极坐标子图并绘制
fig, ax = plt.subplots(subplot_kw={'polar': True})
# pcolormesh的数值参数用占位符，重点设置facecolors为我们的RGB颜色
ax.pcolormesh(angles, radii, np.zeros_like(radii_grid), facecolors=colors, shading='flat')

# 调整半径显示范围
ax.set_ylim(0, 1)
plt.show()

方法二：用坐标变换将笛卡尔图像映射到极坐标 #

如果你的RGB数据本身是笛卡尔格式的图像，想快速转成极坐标显示，可以用坐标变换的方式，把imshow的输出映射到极坐标空间：

import matplotlib.pyplot as plt
import numpy as np
from matplotlib.transforms import Affine2D

data = np.array([[[0,0,1],[0,1,0],[1,0,0]],
                 [[0,0,0.5],[0,0.5,0],[0.5,0,0]]])

fig, ax = plt.subplots(subplot_kw={'polar': True})

# 创建从笛卡尔坐标到极坐标的变换：先缩放再绑定极坐标轴的变换
transform = Affine2D().scale(2*np.pi, 1) + ax.transData
# 绘制图像时指定变换规则
ax.imshow(data, extent=[0,1,0,1], transform=transform, origin='lower')

ax.set_ylim(0,1)
plt.show()

两种方案的差异 #

• 方法一能精确控制每个极坐标网格点的颜色，适合数据本身就是极坐标维度组织的场景；
• 方法二更适合快速转换已有笛卡尔图像，但可能会有轻微的拉伸变形，需要根据实际需求调整参数。

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