Random sampling in numpy | randint() function