相关知识
使用Numpy,我们可以方便的对数组进行拆分,比如使用hsplit()、vsplit()、dsplit()、split()函数等。
拆分数组
准备数组。
import numpy as np
In: a= np.arange(9).reshape(3,3)
In: a
Out: array([[0, 1, 2],
[3, 4, 5],
[6, 7, 8]])
hsplit()
横向拆分。
In: np.hsplit(a,3)
Out:[array([[0],[3], [6]]),
. array([[1], [4],[7]]),
array([[2],[5], [8]])]
vsplit()
纵向拆分。
In: np.vsplit(a,3