Un programa mínimo para dibujar un gráfico con le módulo pyplot es el siguiente:
import matplotlib.pyplot as plt
#plt is shorthand for matplotlib.pyplot
x = (1, 2, 3, 4)
y = (2, 5, 4, 7)
plt.xlabel('x') # introduciong Mr. X-axis label
plt.ylabel('y') # and Ms. Y-axis label
plt.plot(x, y, 'bo') # 'bo' means blue circles
plt.show() # yes, wanna see it on the screen
Un programa algo más complejo que dibuja una función
import numpy as np
import matplotlib.pyplot as plt
def f(x): # the displayed function
return x**2 * np.exp(-x**2)
x = np.arange(0.0, 5.0, 0.01) # a range with decimal increments
y = f(x) # the f(x) values for each x
plt.plot(x, y, 'r-',
label='x^2epx(-x^2)') # 'r-' means red lines
plt.legend() # yes, we want a legend
plt.savefig('f.png') # print to file
plt.show() # please, display
Escribe ahora un programa que dibuje el mejor ajuste lineal a un conjunto de puntos del plano (y los puntos dados). Los datos se leerán de un archivo llamado datos.txt que contiene dos números reales por línea separados por espacios en blanco que representan las coordenadas de cada punto. Si necesitas más información sobre las opciones básicas de pyplot, puedes consultar aquí.
Para determinar el rango en el que se debe calcular y dibujar la recta, se puede utilizar las funciones predefinidas max(X) y min(X) que devuelven, respectivamente, el valor máximo y mínimo de las componentes de una lista o vector X.
Para modificar los límites de la gráfica, se puede utilizar las funciones xlim e ylim:
plt.xlim(xlow, xhigh)# box is [xlow,xhigh] x [ylow,yhigh]
plt.ylim(ylow, yhigh)#
Esta introducción termina con el ejercicio 4 (estructuras de datos).