Numpy中的split()等价于vsplit(),可以使用它来分割矩阵

numpy.split(ary, indices_or_sections, axis=0)

ary:是待分割的矩阵
indices_or_sections:可以传数值或是数组,传数值的话会将矩阵分割成数值的份数,如果传数组则按照数组中的大小区间分割.

>>> x = np.arange(9.0)
>>> np.split(x, 3)
[array([ 0.,  1.,  2.]), array([ 3.,  4.,  5.]), array([ 6.,  7.,  8.])]


>>> x = np.arange(8.0)
>>> np.split(x, [3, 5, 6, 10])
[array([ 0.,  1.,  2.]),
 array([ 3.,  4.]),
 array([ 5.]),
 array([ 6.,  7.]),
 array([], dtype=float64)]
posted @ 2018-05-12 11:08:27
评论加载中...

发表评论