Python中的RGB极坐标图

Question

我正在尝试用Python生成RGB极坐标图，我希望matplotlib.pyplot.imshow能够做到这一点。但是，每当我尝试使用此方法绘制数据时，我都会获得一个空白输出。

import matplotlib.pyplot as plt
import numpy as np

data = np.array([[[0,0,1],[0,1,0],[1,0,0]],[[0,0,0.5],[0,0.5,0],[0.5,0,0]]])
# Sample, any N,M,3 data should work
ax = plt.subplot(111,polar=True)
ax.imshow(data,extent=[0,2*np.pi,0,1]) # Produces a white circle

使用上述方法或其他方法是否有很好的方法来实现这一目标？

感谢。

编辑：我设法使用extent=[0,np.pi/2,0,1]制作了一个象限，但它的使用明显被用于极坐标图。因为除了完整象限以外的任何东西都不会产生预期的结果。

Answer 1

遗憾的是，在极坐标图上使用imshow是不可能的，因为imshow网格的像素必然是二次的。但是，您可以使用pcolormesh并应用技巧（类似于this one），即将color参数的颜色提供给pcolormesh，因为它通常只需要2D输入。

import matplotlib.pyplot as plt
import numpy as np

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

ax = plt.subplot(111, polar=True)

#get coordinates:
phi = np.linspace(0,2*np.pi,data.shape[1]+1)
r = np.linspace(0,1,data.shape[0]+1)
Phi,R = np.meshgrid(phi, r)
# get color
color = data.reshape((data.shape[0]*data.shape[1],data.shape[2]))

# plot colormesh with Phi, R as coordinates, 
# and some 2D array of the same shape as the image, except the last dimension
# provide colors as `color` argument
m = plt.pcolormesh(Phi,R,data[:,:,0], color=color, linewidth=0)
# This is necessary to let the `color` argument determine the color
m.set_array(None)


plt.show()

结果不是圆圈，因为您没有足够的积分。重复数据后，data = np.repeat(data, 25, axis=1)将允许获得一个圆圈。