python animation极坐标动态图_Python中的RGB极坐标图

不幸的是，在极坐标图上使用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)将允许得到一个圆。在