numpy code