Dessin graphique avec matplotlib

J'utilise beaucoup le dessin graphique, donc j'ai personnellement créé le minimum nécessaire. Je vais omettre l'explication des fonctions dans le matplotlib de base, car elles sont probablement expliquées par d'autres. ʻImporter omettre l'importation de matplotlib.pyplot`.

plot_graph.py

def plot_graph(x, y=[], graph_name='graph.png', \
    save_flag=True, show_flag=False, label_x='', label_y='', label='',title=''):
    if label: pyplot.plot(x, y, label=label)
    else: pyplot.plot(x, y)
    if title: pyplot.title(title)
    if label_x: pyplot.xlabel(label_x) 
    if label_y: pyplot.ylabel(label_y)
    if save_flag: pyplot.savefig(graph_name)
    if show_flag: pyplot.show()

Par défaut, si vous donnez x, y comme argument, il sera écrit dans le fichier graph.png. «x, y» est une entrée de chaîne numérique. En option, je mets le nom du fichier, sauvegardé ou non, modifiable pour afficher, chaque étiquette d'axe, étiquette de graphe, titre. J'ai pensé qu'il serait préférable de séparer save_flag et show_flag, mais j'ai pensé que ce serait plus facile à comprendre quand je le verrais plus tard.

Partiellement fonctionnalisé ci-dessous

partie de l'intrigue

plot_only.py

def plot_only(x, y=[], label=''):
    if label: 
        if y: pyplot.plot(x, y, label=label)
        else: pyplot.plot(x, label=label)
    else: 
        if y: pyplot.plot(x, y)
        else: pyplot.plot(x)

Correspondance lorsqu'il y a une étiquette et qu'il n'y a qu'un seul argument

Partie d'étiquetage

add_label.py

def add_label(label_x='', label_y='', title=''):
    if title: pyplot.title(title)
    if label_x: pyplot.xlabel(label_x) 
    if label_y: pyplot.ylabel(label_y)

Partie de sortie

output_graph.py

def output_graph(save_flag=True, show_flag=False, graph_name='graph.png'):
    if save_flag: pyplot.savefig(graph_name)
    if show_flag: pyplot.show()

Maintenant que nous avons une fonction, rendez plot_graph.py intelligent

new_plot_graph.py

def new_plot_graph(x, y=[], graph_name='graph.png', \
    save_flag=True, show_flag=False, label_x='', label_y='', label='',title=''):
    plot_only(x, y, label)
    add_label(label_x, label_y, title)
    output_graph(save_flag, show_flag, graph_name)

Je n'utilise pas tellement la légende, la plage d'affichage, xtrick, etc., donc je ne l'ai pas écrit cette fois. Comme ça, c'était une fonction dont j'avais besoin.