Kfold Cross Validation Uneven Split for Y

Question

I would like to split my data with sklearn.kfold. Here\'s my code

kf = KFold(n_splits = 10)

print(np.shape(X_1))
print(np.shape(y))


for train_index, test_i