def kinetic_energy(vel):
 return (0.5 * (vel ** 2).sum(axis=1))

def kinetic_energy ( vel ) : return ( 0.5 * ( vel ** 2 ) . sum ( axis = 1 ) )
returns the kinetic energy associated with the given velocity and mass of 1 .