当前位置: 首页 > 文档资料 > NumPy 中文教程 >

vsplit

优质
小牛编辑
135浏览
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]])]