xxxxxxxxxx
import numpy as np, random
Unbalanced = [[i, np.random.randint(2)] for i in range(1000)]
Zeros = [i for i in Unbalanced if i[1]==0]
Ones = [i for i in Unbalanced if i[1]==1]
Cut = min([len(Zeros), len(Ones)])
DataSet = Zeros[:Cut] + Ones[:Cut]
random.shuffle(DataSet)
print(f'Unbalanced: {len(Unbalanced)}, Balanced: {len(DataSet)}')