Wave equation in python. .