将一些值绘制到matplotlib中的极坐标图中

时间:2012-10-18 11:08:39

标签: python matplotlib

我想用Python和matplotlib将代表可见卫星的一些值绘制成极坐标图。我在matplotlib示例后面写了一些代码,并且表示了极坐标图,但没有绘制给定的点:

import matplotlib
from matplotlib.pyplot import figure, show, rc, grid
from math import pi

# radar green, solid grid lines
rc('grid', color='#316931', linewidth=1, linestyle='-')
rc('xtick', labelsize=15)
rc('ytick', labelsize=15)

# force square figure and square axes looks better for polar, IMO
width, height = matplotlib.rcParams['figure.figsize']
size = min(width, height)
# make a square figure
fig = figure(figsize=(size, size))
ax = fig.add_axes([0.1, 0.1, 0.8, 0.8], polar=True, axisbg='#d5de9c')

# Satellite info [PRN, E, Az, Ss, Used (1 yes, 0 no)]
sat = [ [1, 62, 255, 46, 1],
        [14, 62, 26, 46, 1],
        [31, 42, 158, 36, 1],
        [22, 40, 76, 50, 1],
        [11, 29, 308, 0, 0],
        [19, 26, 243, 36, 1],
        [3, 13, 217, 0, 0],
        [18, 10, 93, 0, 0],
        [20, 6, 291, 0, 0],
        [5, 1, 72, 0, 0],
        [122, 43, 216, 0, 0],
        [135, 47, 203, 43, 0] ]

for index in (0, len(sat)-1):
    if(sat[index][4]>0):
        ax.plot(sat[index][2], sat[index][1], color='green', marker='s', markersize=12)
    else:
        ax.plot(sat[index][2], sat[index][1], color='gray', marker='s', markersize=12)

ax.set_rmax(2.0)
grid(True)

ax.set_title("Visible satellites", fontsize=20)
show()

我做错了什么?

2 个答案:

答案 0 :(得分:1)

在这一行:

for index in (0, len(sat)-1):

索引仅在一对值上运行。你的意思是for index in range(0, len(sat))

答案 1 :(得分:1)

只需迭代你的sats列表就可以了:

for s in sat:
    if(s[4]>0):
        ax.plot(s[2], s[1],color='green', marker='s', markersize=5)
    else:
        ax.plot(s[2], s[1],color='gray', marker='s', markersize=5)

除此之外,你还使用set_rmax将方位角限制为< 2,这样只留下1个卫星可见,取消注释以查看所有卫星。

编辑: 直接解压缩列表可能会进一步提高可读性:

for (PRN, E, Az, Ss, Used) in sat:
    if(Used>0):
        ax.plot(Ss, Az,color='green', marker='s', markersize=5)
    else:
        ax.plot(Ss, Az,color='gray', marker='s', markersize=5)