Python numpy.hsplit()用法及代碼示例

numpy.hsplit()函數將一個數組水平(按列)拆分為多個子數組。 hsplit等效於使用axis = 1進行拆分，無論數組尺寸如何，數組始終沿第二條軸拆分。

用法： numpy.hsplit(arr, indices_or_sections)

參數：

arr  :[ndarray] Array to be divided into sub-arrays.

indices_or_sections  :[int or 1-D array] If indices_or_sections is an integer, N, the array will be divided into N equal arrays along axis.

If indices_or_sections is a 1-D array of sorted integers, the entries indicate where along axis the array is split

Return :[ndarray] A list of sub-arrays.

代碼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.]]])]