joshcherian42

import pandas as pd
import numpy as np
import time

r = 20
c = 4
groups = [n for n in range(4) for b in range(5)]
data = np.random.randint(0, 10, size=(r, c))
data = np.insert(data, 0, groups, axis=1)