In this example we will show how to split the 2-D NumPy arrays along rows using the array_split() method in Python.

Source Code

import numpy as np

n = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9], [10, 11, 12], [13, 14, 15]])

split_n = np.array_split(n, 3, axis=1)

print(split_n)

Output:

[array([[ 1],
       [ 4],
       [ 7],
       [10],
       [13]]), array([[ 2],
       [ 5],
       [ 8],
       [11],
       [14]]), array([[ 3],
       [ 6],
       [ 9],
       [12],
       [15]])]