Random sampling in numpy | sample() function