Coding in Python

Write a function in python that split the data into k-fold without sklearn. The goal is to split N-dimensional data into K-folds with train and test set.

- Here is the dataset example:

np.array([[1, 2, 3, 4, 5], [6, 7, 8, 9, 10], [11, 12, 13, 14, 15], [16, 17, 18, 19, 20], [21, 22, 23, 24, 25], [26, 27, 28, 29, 30])

- function returns like the image below:

5-fold CV DATASET Estimation 1 Test Train Train Train Train Estimation 2 Train Train Train Train Test Train Estimation 3 Train Test Train Train Estimation 4 Train Train Train Test Train Estimation 5 Train Train Train Train Test