Comment faire pour convertir un fichier CSV à un graphique en Python

Créez un fichier CSV simple pour les tests. Un échantillon pourrait ressembler à ceci:

1,2
2,3
3,8
4,13
5,18
6,21
7,13
7.5,4
2.5,4.3

Importez les bibliothèques python nécessaires dans votre fichier de code:

matplotlib.pyplot d'importation comme plt
csv d'importation
import sys

Ouvrez le fichier CSV et créer un objet lecteur de lui. Déclarer des variables pour définir les limites supérieures et inférieures pour les axes x et y des valeurs du graphe:

csv_reader = csv.reader (open ('test.csv'))
BigX = flotteur (-sys.maxint -1)
bigy = flotteur (-sys.maxint -1)
smallx = flotteur (sys.maxint)
smally = flotteur (sys.maxint)

Itérer sur chaque ligne contenue dans l'objet lecteur stocker chaque ligne comme un sommet dans un tableau de vertex. Dans la même boucle de comparer les valeurs x et y dans le but de stocker leurs limites supérieures et inférieures. Trier à nouveau le tableau de vertex et ensuite en boucle à travers elle. Ce magasin de temps les x triées et valeurs y dans des tableaux distincts:

verts = []
la ligne de la csv_reader:

verts.append (ligne)
si flotteur (row [0]) gt; BigX:
BigX = flotteur (row [0])
si flotteur (row [1]) gt; bigy:
bigy = flotteur (row [1])
si flotteur (row [0]) lt; smallx:
smallx = flotteur (row [0])
si flotteur (row [1]) lt; smally:
smally = flotteur (row [1])

verts.sort ()
x_arr = []
y_arr = []
pour en vert verts:

x_arr.append (vert [0])
y_arr.append (vert [1])

Créer un objet FigureCanvas utilisant l'objet matplotlib de pyplot importé. Ajouter les axes du graphique à FigureCanvas en appelant les add_axes de fonction et en lui passant un tableau de valeurs sous la forme de: gauche, en bas, la largeur, la hauteur. Ces valeurs définissent où le graphique est placé sur la toile --- ils peuvent aller de 0,0 à 1,0:

fig plt.figure = ()
ax = fig.add_axes ([0,1, 0,1, 0,8, 0,8])

Format graphique ajout d'étiquettes et de définir les valeurs minimales et maximales pour chaque axe:

ax.set_xlabel ('x données »)
ax.set_ylabel («données y ')
ax.set_xlim (smallx, BigX)
ax.set_ylim (smally, bigy)

Tracer le graphique en passant dans les deux tableaux contenant les valeurs x et y récupérées à partir du fichier CSV. Personnalisez la parcelle de ligne en passant des valeurs optionnelles telles que la couleur de la ligne (de couleur) ou la largeur de ligne (LW). Afficher le graphique fini par appeler la méthode show pour ouvrir une fenêtre et stocker l'image en appelant savefig pour créer un fichier bitmap sur le disque:

ax.plot (x_arr, y_arr, color = "blue", LW = 2)
plt.show ()
fig.savefig ('test.png')