plot_rader_chart.py 文件源码

def draw_rader_chart_restaurant(lat, lng, df):
    """
    This function call the method in class 'ComplexRadar' to draw the rader chart for each recommend restaurant,
    and create a pdf with restaurants name and rader chart.

    Parameters:
        lat: float
        lng: float
        df: Dataframe

    Return:
        create a pdf file
    """
    variables_restaurant = ('number_of_price', 'Reviews', 'score_of_review', 'Distance')
    ranges_restaurant = [(0.00001, 5), (0.00001, 5), (0.00001, 5), (0.00001, 10)]

    data = list(zip(df['number_of_price'], df['Reviews'], df['Avgscore'], df['Distance']))
    information = list(zip(df['Name']))

    path = os.path.abspath("Results")
    pp = PdfPages(filename = path + '/Recommendation_restaurants.pdf')

    for i in range(len(data)):
        fig = plt.figure(i, figsize=(4, 6))
        rader = ComplexRadar(fig, variables_restaurant, ranges_restaurant)
        rader.plot(data[i])
        rader.fill(data[i], alpha=0.2)
        text = 'Restaurant Name: ' + str(information[i][0])
        fig.text(0, 0.9, text, fontsize=15, fontweight='bold', color = 'blue')
        pp.savefig(bbox_inches = 'tight')
        plt.clf()
    pp.close()

    webbrowser.open_new('file://' + path + '/Recommendation_restaurants.pdf')