Use numpy to solve transport equation with wave-like initial condition Use numpy to solve transport equation with wave-like initial condition numpy numpy