playerDraw.py

import numpy as np
import matplotlib.pyplot as plt
import pandas as pd
import seaborn as sns
from PIL import Image


def draw_radar(arr, feature, title):
    arr = arr/100
    plt.clf()
    plt.rcParams['axes.unicode_minus'] = False
    plt.style.use('seaborn-paper')
    angles = np.linspace(0.3 * np.pi, 2.3 * np.pi, len(feature), endpoint=False)
    values = np.concatenate((arr, [arr[0]]))
    angles = np.concatenate((angles, [angles[0]]))
    feature = np.concatenate((feature, [feature[0]]))
    fig = plt.figure()
    ax = fig.add_subplot(111, polar=True)
    ax.plot(angles, values, 'o-', linewidth=2)
    ax.fill(angles, values, alpha=0.25)
    ax.set_thetagrids(angles * 180 / np.pi, feature)
    ax.tick_params(labelsize=15)
    ax.set_ylim(0, 1)
    ax.set_title(title, fontsize= 30)
    ax.grid(True)
    ax.tick_params('y', labelleft=False)
    plt.savefig('.\\static\\img\\'+title+'.png')


def draw_rating(data):
    par_names = ['STRating', 'LWRating', 'LFRating', 'CFRating', 'RFRating', 'RWRating',
                 'CAMRating', 'LMRating', 'CMRating', 'RMRating', 'LWBRating', 'CDMRating',
                 'RWBRating', 'LBRating', 'CBRating', 'RBRating', 'GKRating']
    df = pd.DataFrame(
        {
            'Name': par_names,
            'Value': data
        })

    # set figure size
    plt.figure(figsize=(10, 10))

    # plot polar axis
    ax = plt.subplot(111, polar=True)

    # remove grid
    plt.axis('off')

    # Set the coordinates limits
    upperLimit = 100
    lowerLimit = 30

    # Compute max and min in the dataset
    max = df['Value'].max()

    # Let's compute heights: they are a conversion of each item value in those new coordinates
    # In our example, 0 in the dataset will be converted to the lowerLimit (10)
    # The maximum will be converted to the upperLimit (100)
    slope = (max - lowerLimit) / max
    heights = slope * df.Value + lowerLimit

    # Compute the width of each bar. In total we have 2*Pi = 360°
    width = 2 * np.pi / len(df.index)

    # Compute the angle each bar is centered on:
    indexes = list(range(1, len(df.index) + 1))
    angles = [element * width for element in indexes]

    # Compute the angle each bar is centered on:
    indexes = list(range(1, len(df.index) + 1))
    angles = [element * width for element in indexes]

    # initialize the figure
    plt.figure(figsize=(6.4, 4.4))
    ax = plt.subplot(111, polar=True)
    plt.axis('off')

    # Draw bars
    bars = ax.bar(
        x=angles,
        height=heights,
        width=width,
        bottom=lowerLimit,
        linewidth=2,
        edgecolor="white",
        color="#61a4b2",
    )

    # little space between the bar and the label
    labelPadding = 4

    # Add labels
    for bar, angle, height, label in zip(bars, angles, heights, df["Name"]):

        # Labels are rotated. Rotation must be specified in degrees :(
        rotation = np.rad2deg(angle)

        # Flip some labels upside down
        alignment = ""
        if angle >= np.pi / 2 and angle < 3 * np.pi / 2:
            alignment = "right"
            rotation = rotation + 180
        else:
            alignment = "left"

        # Finally add the labels
        ax.text(
            x=angle,
            y=lowerLimit + bar.get_height() + labelPadding,
            s=label,
            ha=alignment,
            va='center',
            rotation=rotation,
            rotation_mode="anchor")

    plt.savefig(".\\static\\img\\rating.png")