python

[numpy] numpy 분할 | 열 분할 | 행 분할

전감자(◔◡◔) 2022. 9. 29. 01:21

- 열 분할 (열의 개수가 줄어듦 가로로 짧아짐 => hsplit)

'''
열분할==> 분할 개수가 동일
1) hsplit
2) split( axis=1 )

'''
import numpy as np

arr=np.arange(12).reshape(3,4)
print(arr)
'''
[[ 0  1  2  3]
 [ 4  5  6  7]
 [ 8  9 10 11]]
'''

print(np.hsplit(arr,2))
arr1,arr2=np.hsplit(arr,2)
arr1,arr2=np.split(arr,2,axis=1)
print(arr1,arr2)
'''
[array([[0, 1],
       [4, 5],
       [8, 9]]), 
array([[ 2,  3],
       [ 6,  7],
       [10, 11]])]
       
       
[[0 1]
 [4 5]
 [8 9]]
  
  [[ 2  3]
 [ 6  7]
 [10 11]]

'''

 

- 행 분할 (행의 개수가 줄어듦 세로로 짧아짐 vsplit)

'''
행분할==> 분할 개수가 동일
1) vsplit
2) split( axis=1 )

'''
import numpy as np

arr=np.arange(12).reshape(4,3)
print(arr)
'''
[[ 0  1  2]
 [ 3  4  5]
 [ 6  7  8]
 [ 9 10 11]]
'''
arr1,arr2=np.vsplit(arr,2)
arr1,arr2=np.split(arr,2,axis=0)
print(arr1,arr2)

'''
[[0 1 2]
 [3 4 5]]
  
  [[ 6  7  8]
 [ 9 10 11]]

'''