numpy pythagorean theorem