vsplit
优质
小牛编辑
140浏览
2023-12-01
numpy.vsplit是split()函数的一个特例,其中axis为1表示垂直分割,与输入数组的维度无关。 以下示例清楚地说明了这一点。
例子 (Example)
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]])]