Python - Crear animación

for resultado in resultados:

    graficar(nodos, celdas, resultado)

plt.show()

def graficar(nodos, celdas, resultados):

    # Obtencion de las coordenadas x,y de los nodos -> puntos, x, y

    # Transformacion de los elementos de la malla a numero -> triangles

    # Triangulación de los nodos para crear los elementos segun el criterio de triangles

    puntos = np.asfarray(nodos, float)

    x = puntos[:, 0]

    y = puntos[:, 1]

    triangles = celdas.astype(np.float) - 1

    triang = mtri.Triangulation(x, y, triangles)

    # Transformacion de los resultados en un vector -> z

    z = [resultado[0] for resultado in resultados]

    # Graficar los elementos y los contornos

    plt.figure()

    tpc = plt.tripcolor(triang, z, cmap=cm.jet, edgecolors='k')

    plt.colorbar(tpc)

    plt.title('Evolución de T [K] con {} elementos'.format(len(triangles)))

    plt.xlabel('X [m]')

    plt.ylabel('Y [m]')