numpy random sample first axis code example Example: numpy generate random 2d array import numpy as np a=np.random.rand(3,3) print(a)