import numpy import math def modulus(vec): tmp = 0 for comp in vec: tmp += comp**2 return math.sqrt(tmp)