Python numpy.hsplit()
numpy.hsplit()函数将一个数组横向(column-wise)分割成多个子数组。hsplit等同于axis=1的split,无论数组维度如何,数组总是沿着第二轴分割。
语法: numpy.hsplit(arr, indices_or_sections)
参数 :
arr : [ndarray] 阵列将被划分为子阵列。
indices_or_sections : [int or 1-D array] 如果indices_or_sections是一个整数,N,数组将沿轴线被分成N个相等的数组。
如果indices_or_sections是一个排序的整数的一维数组,那么这些条目表示该数组沿轴线的分割位置
返回: [ndarray] 一个子数的列表。
代码#1:
# Python program explaining
# numpy.hsplit() function
# importing numpy as geek
import numpy as geek
arr = geek.arange(16.0).reshape(4, 4)
gfg = geek.hsplit(arr, 2)
print (gfg)
输出 :
[array([[ 0., 1.],
[ 4., 5.],
[ 8., 9.],
[ 12., 13.]]), array([[ 2., 3.],
[ 6., 7.],
[ 10., 11.],
[ 14., 15.]])]
代码#2:
# Python program explaining
# numpy.hsplit() function
# importing numpy as geek
import numpy as geek
arr = geek.arange(27.0).reshape(3, 3, 3)
gfg = geek.hsplit(arr, 1)
print (gfg)
输出 :
[array([[[ 0., 1., 2.],
[ 3., 4., 5.],
[ 6., 7., 8.]],
[[ 9., 10., 11.],
[ 12., 13., 14.],
[ 15., 16., 17.]],
[[ 18., 19., 20.],
[ 21., 22., 23.],
[ 24., 25., 26.]]])]