在极轴上对两条线之间的线段进行着色(matplotlib)

时间:2018-02-16 15:50:10

标签: python matplotlib

目前,我有一个图表,其中的线条表示日出(蓝色),中午(红色),日落(橙色)和对象(绿色)的方向。我想在日出和日落之间创建一个黄色阴影区域。我想到了plt.fill方法,但是当我添加plt.fill_between(theta1, theta2, R1, alpha=0.2)时,这似乎没有用。我认为这是因为两条线不相交。我目前的极轴图看起来像:

enter image description here

脚本的相关部分是:

    #for object
    theta0 = np.deg2rad([190, 190])
    print ("theta0= %s" %theta0)    
    R0 = [0,0.4]

    #for sunrise
    theta1 = np.deg2rad([105, 105])
    print ("theta1= %s" %theta1)    
    R1 = [0,1]

    #for sunset
    theta2 = np.deg2rad([254, 254])
    print ("theta2= %s" %theta2)    
    R1 = [0,1]

    #for midday 
    theta3 = np.deg2rad([0.2, 0.2])
    print ("theta3= %s" %theta3)    
    R3 = [0,1]

    ax.set_theta_zero_location("S")
    ax.set_theta_direction(-1)

    ax.plot(theta1,R1, theta2, R1, theta0, R0, lw=5)

    #plt.fill_between(theta1, theta2, R1, alpha=0.2) #does not work


    plt.savefig('plot.png')

我还想过使用中午的方位角,然后使用宽度,根据日落和日出之间的角度差计算,然后使用这些来生成阴影区域。我怎么能这样做呢?我想在日出和日落之间只有一个黄色的阴影区域,因此我认为从正午的theta产生width将是一个好主意。我尝试复制ax以形成ax2,然后我可以使用它来显示正午的width(阴影区域)。但随着ax2的加入,它似乎搞砸了情节。我怎么解决这个问题呢?

1 个答案:

答案 0 :(得分:2)

要在极坐标图上使用fill_between,您必须将其称为

plt.fill_between(angles_to_go_through, minimum_radii, maximum_radii)

因此,例如,在0到90度之间填充

ax.fill_between(
    np.linspace(0, pi/2, 100),  # Go from 0 to pi/2
    0,                          # Fill from radius 0
    1,                          # To radius 1
)

这会产生类似

的情节

polar fill

在您的情况下,您可以致电

ax.fill_between(
    np.linspace(theta1[0], theta2[0], 100),  # Need high res or you'll fill a triangle
    0,
    1,
    alpha=0.2,
    color='y',
)