Python pyplot.rgrids函数代码示例

def plot_radar(example_data,nVar):
    N = nVar
    theta = radar_factory(N, frame='polygon')

    data = example_data
    people_Num=len(data)
    spoke_labels = data.pop('column names')

    fig = plt.figure(figsize=(9, 2*people_Num))
    fig.subplots_adjust(wspace=0.55, hspace=0.10, top=0.95, bottom=0.05)

    colors = ['b', 'r', 'g', 'm', 'y']
    for n, title in enumerate(data.keys()):
        ax = fig.add_subplot(int(people_Num/3)+1, 3, n+1, projection='radar')
        plt.rgrids([0.2, 0.4, 0.6, 0.8])
        plt.setp(ax.get_yticklabels(), visible=False)
        plt.ylim([0,1]) 
        ax.set_title(title, weight='bold', size='medium', position=(0.5, 1.1),color='b',
                     horizontalalignment='center', verticalalignment='center',fontproperties=zhfont)
        for d, color in zip(data[title], colors):
            ax.plot(theta, d, color=color)
            ax.fill(theta, d, facecolor=color, alpha=0.25)
        ax.set_varlabels(spoke_labels)

    plt.subplot(int(people_Num/3)+1, 3, 1)

    plt.figtext(0.5, 0.965, '战力统计',fontproperties=zhfont,
                ha='center', color='black', weight='bold', size='large')
    plt.show()

    def plot(self, legend=True, fig_title=None):
        theta = radar_factory(self.get_data_size(), frame="polygon")

        fig = plt.figure(figsize=(9, 9))
        fig.subplots_adjust(wspace=0.25, hspace=0.20, top=0.85, bottom=0.05)

        colors = ["b", "r", "g", "m", "y"]
        # Plot the four cases from the example data on separate axes
        for n, title in enumerate(self._data.keys()):
            ax = fig.add_subplot(2, 2, n + 1, projection="radar")
            plt.rgrids([0.02, 0.04, 0.06, 0.08])
            ax.set_title(
                title,
                weight="bold",
                size="medium",
                position=(0.5, 1.1),
                horizontalalignment="center",
                verticalalignment="center",
            )
            for d, color in zip(self._data[title], colors):
                ax.plot(theta, d, color=color)
                ax.fill(theta, d, facecolor=color, alpha=0.25)
            ax.set_varlabels(self._labels)

        if legend:
            self.plot_legend()
        if fig_title is not None:
            self.plot_title(fig_title)
        plt.show()

def radar_plot():
    """
    radar plot
    """
    # 生成测试数据
    labels = np.array(["A组", "B组", "C组", "D组", "E组", "F组"])
    data = np.array([68, 83, 90, 77, 89, 73])
    theta = np.linspace(0, 2*np.pi, len(data), endpoint=False)

    # 数据预处理
    data = np.concatenate((data, [data[0]]))
    theta = np.concatenate((theta, [theta[0]]))

    # 画图方式
    plt.subplot(111, polar=True)
    plt.title("雷达图", fontproperties=myfont)

    # 设置"theta grid"/"radar grid"
    plt.thetagrids(theta*(180/np.pi), labels=labels, fontproperties=myfont)
    plt.rgrids(np.arange(20, 100, 20), labels=np.arange(20, 100, 20), angle=0)
    plt.ylim(0, 100)

    # 画雷达图,并填充雷达图内部区域
    plt.plot(theta, data, "bo-", linewidth=2)
    plt.fill(theta, data, color="red", alpha=0.25)

    # 图形显示
    plt.show()
    return

def plot_graph(var_stats, var_clean, var_evil):

    try:
        import matplotlib.pyplot as plt       
    except:
        sys.sderr.write("matploitlib not found")
        sys.exit(1)

    N = 4 # az AZ 09 other
    theta = radar_factory(N, frame='circle')
    spoke_labels = ["[a-z]", "[A-Z]", "[0-9]", "[other]"]

    fig = plt.figure(figsize=(15, 8))
    fig.subplots_adjust(wspace=0.20, hspace=0.20, top=1.00, bottom=0.00, left=0.05, right=0.93)

    """
    # Overall
    ax = fig.add_subplot(2, 2, 3, projection='radar')
    plt.rgrids([20, 40, 60, 80])

    ax.set_title("Overall", weight='bold', size='medium', position=(0.5, 1.1), horizontalalignment='center', verticalalignment='center')

    for varname, varfeatures in var_stats.iteritems():
        d = [ varfeatures['az'], varfeatures['AZ'], varfeatures['09'], varfeatures['other'] ]
        color = 'green' if varname in var_clean else 'red'
        ax.plot(theta, d, color=color)
        ax.fill(theta, d, facecolor=color, alpha=0.25)
    ax.set_varlabels(spoke_labels)
    """

    # Clean
    ax = fig.add_subplot(1, 2, 1, projection='radar')
    plt.rgrids([20, 40, 60, 80])

    ax.set_title("Clean", weight='bold', size='medium', position=(0.5, 1.1), horizontalalignment='center', verticalalignment='center')

    for varname, varfeatures in var_clean.iteritems():
        d = [ varfeatures['az'], varfeatures['AZ'], varfeatures['09'], varfeatures['other'] ]
        ax.plot(theta, d, color='green')
        ax.fill(theta, d, facecolor='green', alpha=0.25)
    ax.set_varlabels(spoke_labels)

    # Evil
    ax = fig.add_subplot(1, 2, 2, projection='radar')
    plt.rgrids([20, 40, 60, 80])

    ax.set_title("Evil", weight='bold', size='medium', position=(0.5, 1.1), horizontalalignment='center', verticalalignment='center')

    for varname, varfeatures in var_evil.iteritems():
        d = [ varfeatures['az'], varfeatures['AZ'], varfeatures['09'], varfeatures['other'] ]
        ax.plot(theta, d, color='red')
        ax.fill(theta, d, facecolor='red', alpha=0.25)
    ax.set_varlabels(spoke_labels)

    plt.figtext(0.5, 0.965, 'Variable/Function name features distribution',
                ha='center', color='black', weight='bold', size='large')
    plt.figtext(0.5, 0.935, 'PRICK v0.1',
                ha='center', color='black', weight='normal', size='medium')

    plt.show()

def create_plot():
    N = 9
    theta = radar_factory(N, frame='circle')

    data = example_data()
    spoke_labels = data.pop(0)

    fig = plt.figure(figsize=(9, 9))
    fig.subplots_adjust(wspace=0.25, hspace=0.20, top=0.85, bottom=0.05)

    colors = ['b', 'r', 'g', 'm', 'y']
    # Plot the four cases from the example data on separate axes
    for n, (title, case_data) in enumerate(data):
        ax = fig.add_subplot(2, 2, n + 1, projection='radar')
        plt.rgrids([0.2, 0.4, 0.6, 0.8])
        ax.set_title(title, weight='bold', size='medium', position=(0.5, 1.1),
                     horizontalalignment='center', verticalalignment='center')
        for d, color in zip(case_data, colors):
            ax.plot(theta, d, color=color)
            ax.fill(theta, d, facecolor=color, alpha=0.25)
        ax.set_varlabels(spoke_labels)

    # add legend relative to top-left plot
    plt.subplot(2, 2, 1)
    labels = ('Factor 1', 'Factor 2', 'Factor 3', 'Factor 4', 'Factor 5')
    legend = plt.legend(labels, loc=(0.9, .95), labelspacing=0.1)
    plt.setp(legend.get_texts(), fontsize='small')

    plt.figtext(0.5, 0.965, '5-Factor Solution Profiles Across Four Scenarios',
                ha='center', color='black', weight='bold', size='large')
    plt.show()

def subplot(data, spoke_labels, sensor_labels,saveto=None,frame_type='polygon'):
    #def subplot(data, spoke_labels, sensor_labels,saveto=None,frame_type='polygon'):
    num_of_picks=9
    theta = radar_factory(len(spoke_labels), frame='circle')
    fig = plt.figure(figsize=(num_of_picks, num_of_picks))
    fig.subplots_adjust(wspace=0.25, hspace=0.20, top=0.85, bottom=0.05)
    num_col=np.floor(np.sqrt(len(data)))
    num_row=np.ceil(num_of_picks/num_col)
    for k,(data_col,sensor_label_) in enumerate(zip(data,sensor_labels)):
        #subplot(num_col,num_row,i+1)
        ax = fig.add_subplot(num_col,num_row,k+1, projection="radar")
        ax.plot(theta, data_col)
        ax.fill(theta, data_col, alpha=0.2)
        ax.set_varlabels(spoke_labels)
        #plt.title(sensor_label_,fontsize='small')
        legend = plt.legend([sensor_label_], loc=(-0.2, 1.1), labelspacing=0.01)
        plt.setp(legend.get_texts(), fontsize='small')
        radar_bnd=max(max(data))
        #import pdb;pdb.set_trace()
        rgrid_spacing=np.round(list(np.arange(0.1,radar_bnd,float(radar_bnd)/5)),2)
        plt.rgrids(rgrid_spacing)
        #plt.rgrids([0.1 + 2*i / 10.0 for i in range(radar_bnd)])
        ##radar_chart.plot(data_col, spoke_labels, sensor_label_, saveto="time_radar.png",frame_type='circle')    
    if saveto != None:
        plt.savefig(saveto)        

def plot_radar(data, titles, title='', legends=None, normalize=False, colors=None, fill=True):
    r = data.copy()
    if normalize:
        r.loc[:, titles] = MaxAbsScaler().fit_transform(r.loc[:, titles])
    case_data = r
    case_data.reset_index(inplace=True)
    theta = radar_factory(len(titles), frame='polygon')

    fig = plt.figure(figsize=(7, 7))
    fig.subplots_adjust(wspace=0.25, hspace=0.20, top=0.85, bottom=0.05)

    if colors is None:
        colors = "bgrcmykw"

    ax = fig.add_subplot(1, 1, 1, projection='radar')
    plt.rgrids([0.2, 0.4, 0.6, 0.8])
    ax.set_title(title, weight='bold', size='medium', position=(0.5, 1.1),
                 horizontalalignment='center', verticalalignment='center')
    for d, color in zip(case_data[titles].values, colors):
        ax.plot(theta, d, color=color)
        if fill:
            ax.fill(theta, d, facecolor=color, alpha=0.25)
    # ax.set_rmax(1.0)
    ax.set_varlabels(titles)

    # add legend relative to top-left plot
    if legends:
        plt.subplot(1, 1, 1)
        labels = legends
        legend = plt.legend(labels, loc=(0.9, .95), labelspacing=0.2)
        plt.setp(legend.get_texts(), fontsize=15)
        plt.setp(legend.get_lines(), linewidth=13, alpha=0.50)

    plt.show()

def sonar_graph(ping_readings):
	#print "ping reading:", ping_readings
	#print type(ping_readings[1])
	# force square figure and square axes looks better for polar, IMO
	fig = figure(figsize=(3.8,3.8))
	ax = P.subplot(1, 1, 1, projection='polar')
	P.rgrids([28, 61, 91])
	ax.set_theta_zero_location('N')
	ax.set_theta_direction(-1)

	try:
		theta = 346
		angle = theta * np.pi / 180.0
		radii = [ping_readings[0]]
		width = .15
		bars1 = ax.bar(0, 100, width=0.001, bottom=0.0)
		#print "theta, radii, width: ", theta, radii, width
		bars = ax.bar(angle, radii, width=width, bottom=0.0, color='blue')
		theta = 6
		angle = theta * np.pi / 180.0
		radii = [ping_readings[1]]
		width = .15
		bars = ax.bar(angle, radii, width=width, bottom=0.0, color='blue')	
		theta = 86
		angle = theta * np.pi / 180.0
		radii = [ping_readings[2]]
		width = .15
		bars = ax.bar(angle, radii, width=width, bottom=0.0, color='blue')
		theta = 266
		angle = theta * np.pi / 180.0
		radii = [ping_readings[3]]
		width = .15
		bars = ax.bar(angle, radii, width=width, bottom=0.0, color='blue')
		img_to_return = fig2img(fig)
		P.close(fig)
		return img_to_return 
	except:
		print "Sonar data error... can't graph"
		pass
	#print "finshed graph"
	#pil_img = fig2img(fig)
	#sonar_image = pil_img
	#print type(pil_img), pil_img
	#sonar_image = PILtoCV_4Channel(pil_img)
	#cv.ShowImage("Sonar", sonar_image )
	#cv.MoveWindow ('Sonar',50 ,50 )
	#time.sleep(.01)
	#cv.WaitKey(10)
	#enable line below to make basestation work 12/13/2012
	#fig.savefig('sonar_image.png')

	#Image.open('sonar_image.png').save('sonar_image.jpg','JPEG')
	#print "finished saving"
	#stop
	#garbage cleanup
	#fig.clf()
	
	#gc.collect()
	#del fig
	P.close(fig)

def buildRadar(data, name):
    N = 7
    theta = radar_factory(N, frame="polygon")
    spoke_labels = data.pop(0)

    fig = plt.figure(figsize=(7, 7))
    fig.subplots_adjust(wspace=0.25, hspace=0.20, top=0.85, bottom=0.05)

    colors = ["b", "r", "g", "m", "y"]
    # Plot the four cases from the example data on separate axes
    for n, (title, case_data) in enumerate(data):
        ax = fig.add_subplot(2, 2, n + 1, projection="radar")
        plt.rgrids([0.2, 0.4, 0.6, 0.8])
        ax.set_title(
            title,
            weight="bold",
            size="medium",
            position=(0.5, 1.1),
            horizontalalignment="center",
            verticalalignment="center",
        )
        for d, color in zip(case_data, colors):
            ax.plot(theta, d, color=color)
            ax.fill(theta, d, facecolor=color, alpha=0.25)
        ax.set_varlabels(spoke_labels)

    # add legend relative to top-left plot
    plt.subplot(2, 2, 1)
    labels = ("Factor 1", "Factor 2", "Factor 3", "Factor 4", "Factor 5")
    plt.figtext(0.5, 0.965, "TITLE", ha="center", color="black", weight="bold", size="large")
    savefig(name)
    plt.clf()

def overlaid_plot(terms, ICA_component_number, savepath=None, filtered_num=False):
    """
    Args:
        - ICA_component_number: integer that corresponds to ICA analysis and directory
        - savepath: just shows if no path is provided
        - filtered_num: Integer that corresponds to the number of terms desired in 
                    the radar plot.
    """
    ICA_path = '/Volumes/Huettel/KBE.01/Analysis/Neurosynth/ICA/ICA%s' %ICA_component_number

    big_list = get_all_term_weights(terms, ICA_path)

    if filtered_num:
        big_list = term_weight_filter(big_list, filtered_num)
        N = filtered_num
    else:
        N = ICA_component_number

    data = convert_final_data(big_list)
    
    theta = radar_factory(N, frame='polygon')

    spoke_labels = data.pop('column names')

    fig = plt.figure(figsize=(9, 9))
    fig.subplots_adjust(wspace=0.25, hspace=0.20, top=0.85, bottom=0.05)

    colors = ['b', 'r', 'g', 'y', 'm']
    # Plot the four cases from the example data on separate axes
    for n, title in enumerate(data.keys()):
        ax = fig.add_subplot(1, 1, n+1, projection='radar')
        plt.rgrids([2, 4, 6, 8])
        #plt.rgrids([0.2, 0.4, 0.6, 0.8])
        ax.set_title(title, weight='bold', size='medium', position=(0.5, 1.1),
                     horizontalalignment='center', verticalalignment='center')
        for d, color in zip(data[title], colors):
            ax.set_ylim(0,10)
            ax.plot(theta, d, color=color)
            ax.fill(theta, d, facecolor=color, alpha=0.25)
        ax.set_varlabels(spoke_labels)

    # add legend relative to top-left plot
    plt.subplot(1, 1, 1)
    labels = tuple(terms)
    legend = plt.legend(labels, loc=(0.9, .95), labelspacing=0.1)
    plt.setp(legend.get_texts(), fontsize='large')
    plt.figtext(0.5, 0.965, '',
                ha='center', color='black', weight='bold', size='large')

    if savepath == None:
        plt.show()
    else:
        plt.savefig(savepath)

def initialisation_graphe(ondes,enonce=True):
    plt.clf()            # Nettoyage, on commence un nouveau graphe
    plt.axes(polar=True) # On initie un graphe en polaires
    # Puis on définit le titre
    if enonce: titre = 'Enonce: Sommer '
    else     : titre = 'Corrige: Sommer '
    titre += ", ".join(['${}$'.format(o) for o in ondes[:-1]])
    titre += " et ${}$.".format(ondes[-1])
    plt.title(titre)
    # et finalement les grilles en distances
    plt.rgrids([i+1 for i in range(max_size)])
    plt.thetagrids([i*15 for i in range(360//15)]) # et en angles

def draw_radar(weapons, color_cluster, fitness_cluster, num_samples):

    N = 10
    theta = radar_factory(N, frame='polygon')

    data = get_data(weapons)
    spoke_labels = data.pop('column names')

    fig = plt.figure(figsize=(16, 9))
    fig.subplots_adjust(wspace=0.50, hspace=0.25)

    colors = ['b', 'r', 'g', 'm', 'y']

    weapons = data['Weapon1'], data['Weapon2']

    for i in range(len(weapons)) :
        ax = fig.add_subplot(2, 3, i+1, projection='radar')
        plt.rgrids([0.5], (''))
        ax.set_title("Weapon" + str(i+1), weight='bold', size='medium', position=(0.5, 1.1),
                     horizontalalignment='center', verticalalignment='center')
        ax.plot(theta, weapons[i], color=color_cluster)
        ax.fill(theta, weapons[i], facecolor=color_cluster, alpha=0.25)
        ax.set_varlabels(spoke_labels)

    ax = fig.add_subplot(2, 3, 4)

    plt.ylim(0, 3)
    ax.set_title("Balance")
    ax.boxplot( [fitness_cluster[i][0] for i in range(len(fitness_cluster))] )

    ax = fig.add_subplot(2, 3, 5)

    plt.ylim(0, 2500)
    ax.set_title("Distance")
    ax.boxplot( [fitness_cluster[i][1] for i in range(len(fitness_cluster))] )

    ax = fig.add_subplot(2, 3, 6)

    plt.ylim(0, 20)
    ax.set_title("Kill Streak")
    ax.boxplot( [fitness_cluster[i][2] for i in range(len(fitness_cluster))] )

    # add legend relative to top-left plot
    '''
    plt.subplot(2, 2, 1)
    labels = ('Factor 1', 'Factor 2', 'Factor 3', 'Factor 4', 'Factor 5')
    legend = plt.legend(labels, loc=(0.9, .95), labelspacing=0.1)
    plt.setp(legend.get_texts(), fontsize='small')
    '''

    plt.figtext(0.5, 0.965, 'Mean of clustered weapon, samples = ' + str(num_samples),
                ha='center', color='black', weight='bold', size='large')

def create_figure(all_data): # takes in data and title and creates and saves plot

    width = 0.45  # width of the bars (in radians)

    # create the figure, dont change
    fig = plt.figure()
    ax = fig.add_subplot(111, polar=True)

    # angle positions, 0 to 360 with increments of 360/5
    xo = list(range(0, 360, 360 / 5))
    # Convert to radians and subtract half the width
    # of a bar to center it.
    x = [i * pi / 180 for i in xo]

    # set the labels for each bar, do not change
    ax.set_xticks(x)
    ax.set_xticklabels(['Military\nProwess', 'Productivity', 'Resource', 'Self-\nSufficiency', 'Morale'])
    ax.set_thetagrids(xo, frac=1.15) # frac changes distance of label from circumference of circle

    plt.ylim(0, 100) # sets range for radial grid

    fig.suptitle("India \n1993-2012", fontsize=20, y=0.5, x=0.1) # title of plot

    plt.rgrids([20, 40, 60, 80, 100], angle=33, fontsize=10) # the numbers you see along radius, angle changes position

    colorList = [];

    count = -1
    for key in all_data:
        count = count + 1
        data = all_data[key]
        mylist = [item+0.5*(count-len(all_data)/2)/len(all_data) for item in x]
        bars = ax.bar(mylist, data, width=width, align='center') # do the plotting
        i = 0
        for r, bar in zip(data, bars):
            bar.set_facecolor( cm.jet(0.8*count/len(all_data))) # set color for each bar, intensity proportional to height of bar
            colorList.append(cm.jet(0.8*count/len(all_data)))
            #bar.set_alpha(0.2) # make color partly transparent
            
            height = bar.get_height() # this is basically the radial height, or radius of bar

            # write value of each bar inside it
            # first param is angle, second is radius -10 makes it go inside the bar
        
            if i == 3 and count == 0:
                ax.text(mylist[i]-width/4*3, height+5, key, ha='center', va='center', fontsize=11)
            if i == 3 and count == len(all_data)-1:
                ax.text(mylist[i]+width/4*3, height-5, key, ha='center', va='center', fontsize=11)
            i = i + 1

    
    plt.savefig('examples/multiple.png')

def main():
    azi = wd_11_12
    z = ws_11_12

    plt.figure(figsize=(5,6))
    plt.subplot(111, projection='polar')
    coll = rose(azi, z=z, bidirectional=True)
    plt.xticks(np.radians(range(0, 360, 45)), 
               ['N', 'NE', 'E', 'SE', 'S', 'SW', 'W', 'NW'])
    plt.colorbar(coll, orientation='horizontal')
    plt.xlabel('2011 - 2012 3m Wind rose colored by mean wind speed')
    plt.rgrids(range(5, 20, 5), angle=290)

    plt.savefig('/home/cparr/Snow_Patterns/figures/wind/winter_11_12.png',dpi = 300)

    plt.show()

def plot_phaseplot_l(dictdata, keys, autok, title, withn='yes'):
    """Plots a phase plot where the radius is not fixed to 1."""
    
    colors = ['r', 'b', 'g', 'y', 'k']
    
    plt.suptitle(title, fontsize='large' )
    if autok == 'yes':
        k = dictdata.keys()
    
    for i, condition in enumerate(keys):
        datac = colors[i]
        data = dictdata[condition]
        
        try:
            n = len(data)
            theta, r = zip(*data)
        except TypeError:
            theta, r = data
            n = 1
        if withn == 'yes':
            plt.polar(theta, r, 'o', color=datac, label=condition + '\n n=' + str(n))
        if withn == 'no':
            plt.polar(theta, r, 'o', color=datac, label=condition)

        lines, labels = plt.rgrids( (1.0, 1.4), ('', ''), angle=0 )
        tlines, tlabels = plt.thetagrids( (0, 90, 180, 270), ('0', 'pi/2', 'pi', '3pi/2') )
        leg = plt.legend(loc=(0.93,0.8))
  
        for t in leg.get_texts():
            t.set_fontsize('small')
            
        plt.subplots_adjust(top=0.85)
        plt.draw()