numpy.vsplit是split()函数的特例,其中axis为1表示垂直拆分,而与输入数组的维度无关。以下示例清楚地说明了这一点。
import numpy as np a=np.arange(16).reshape(4,4) print 'First array:' print a print '\n' print 'Vertical splitting:' b=np.vsplit(a,2) print b
First array: [[ 0 1 2 3] [ 4 5 6 7] [ 8 9 10 11] [12 13 14 15]] Vertical splitting: [array([[0, 1, 2, 3], [4, 5, 6, 7]]), array([[ 8, 9, 10, 11], [12, 13, 14, 15]])]
祝学习愉快!(内容编辑有误?请选中要编辑内容 -> 右键 -> 修改 -> 提交!)