numpy 2d array random code example

Example 1: random matrix python

np.random.rand(3,2)

Example 2: np.random.rand()

# train test split
df = pd.read_csv('file_location')
mask = np.random.rand(len(df)) < 0.8
train  = df[mask]
test = df[~mask]