Note
Go to the end to download the full example code.
12.3.10.2.8. Splitting array#
import numpy as np
rg = np.random.default_rng(1)
a = np.floor(10 * rg.random((2, 12)))
a
array([[5., 9., 1., 9., 3., 4., 8., 4., 5., 0., 7., 5.],
[3., 7., 3., 4., 1., 4., 2., 2., 7., 2., 4., 9.]])
b = np.hsplit(a, 3)
b
[array([[5., 9., 1., 9.],
[3., 7., 3., 4.]]), array([[3., 4., 8., 4.],
[1., 4., 2., 2.]]), array([[5., 0., 7., 5.],
[7., 2., 4., 9.]])]
c = np.hsplit(a, (3, 4))
c
[array([[5., 9., 1.],
[3., 7., 3.]]), array([[9.],
[4.]]), array([[3., 4., 8., 4., 5., 0., 7., 5.],
[1., 4., 2., 2., 7., 2., 4., 9.]])]