Python极坐标热力图0与2π处收敛问题咨询

问题说明 #

我用Python写了一段代码，要根据R/theta位置对应的厚度值绘制极坐标热力图，但图表没法正确收敛，不确定是不是要重复0或2π处的值，或者有没有其他解决办法。

原始代码 #

# 注：原代码里的"import libraries"无效，已移除
import numpy as np
import matplotlib.pyplot as plt

z = input("paste the z values from One Note")

thick = [float(num) for num in z.split()]

radius = [500,500,500,500,500,500,500,500,250,250,250,250,250,250,250,250,0]

theta = [1.570796327,
         0.785398163,
         6.283185307,
         5.497787144,
         4.71238898,
         3.926990817,
         3.141592654,
         2.35619449,
         1.570796327,
         0.785398163,
         6.283185307,
         5.497787144,
         4.71238898,
         3.926990817,
         3.141592654,
         2.35619449,
         6.283185307
]

data = [radius,theta,thick]

rad = data[1]
azm = data[0]
z = data[2]

plt.subplot(projection="polar")
plt.tricontourf(rad, azm, z, levels=256, cmap='coolwarm')
plt.colorbar()
plt.show()

问题解决步骤 #

1. 核心错误：坐标参数搞反了 #

极坐标下plt.tricontourf的参数顺序是theta（角度）在前，radius（半径）在后，你代码里把两者映射反了，这是图表异常的主要原因。

2. 0/2π边界连续性处理 #

极坐标里0和2π是同一个位置，要让图表收敛，需要保证边界数据一致：

• 可以把所有2π（即6.283185307）替换为0，避免重复识别为不同位置
• 如果0和2π对应厚度值有差异，建议统一为其中一个数值，或者补充对应点的匹配数据

修正后的代码 #

import numpy as np
import matplotlib.pyplot as plt

z = input("paste the z values from One Note")
thick = [float(num) for num in z.split()]

radius = [500,500,500,500,500,500,500,500,250,250,250,250,250,250,250,250,0]
# 将所有2π替换为0，保证边界连续性
theta = [1.570796327,
         0.785398163,
         0,
         5.497787144,
         4.71238898,
         3.926990817,
         3.141592654,
         2.35619449,
         1.570796327,
         0.785398163,
         0,
         5.497787144,
         4.71238898,
         3.926990817,
         3.141592654,
         2.35619449,
         0
]

# 修正参数顺序：theta在前，radius在后
plt.subplot(projection="polar")
plt.tricontourf(theta, radius, thick, levels=256, cmap='coolwarm')
plt.colorbar()
plt.show()

额外优化建议 #

• 用np.pi系列常量替代硬编码数值，比如np.pi/2对应1.5708，2*np.pi对应6.2832，代码更易读且准确
• 如果数据是按固定角度间隔采集的，建议用np.meshgrid生成规则网格，再用contourf绘制，图表会更平滑

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