어레이 쪼개기

hsplit() 을 이용해서 수평 방향으로 어레이를 쪼갤 수 있습니다.

동일한 형태로 나눠질 어레이의 개수를 지정하거나, 어떤 열(column)에서 나눌지를 지정할 수 있습니다.

동일한 형태로 나누기

import numpy as np

a = np.floor(10*np.random.random((2, 12)))

print(a)
print(np.hsplit(a, 3))

[[5. 7. 6. 5. 4. 6. 4. 8. 9. 3. 7. 5.]
 [5. 9. 0. 0. 0. 8. 7. 8. 9. 7. 4. 7.]]

[array([[5., 7., 6., 5.],
       [5., 9., 0., 0.]]), array([[4., 6., 4., 8.],
       [0., 8., 7., 8.]]), array([[9., 3., 7., 5.],
       [9., 7., 4., 7.]])]

np.hsplit(a, 3) 은 어레이 a를 3개의 동일한 형태의 어레이로 나눈 어레이를 반환합니다.

나눌 열(column) 지정하기

import numpy as np

a = np.floor(10*np.random.random((2, 12)))

print(a)
print(np.hsplit(a, (3, 4)))

[[5. 7. 6. 5. 4. 6. 4. 8. 9. 3. 7. 5.]
 [5. 9. 0. 0. 0. 8. 7. 8. 9. 7. 4. 7.]]

[array([[5., 7., 6.],
       [5., 9., 0.]]), array([[5.],
       [0.]]), array([[4., 6., 4., 8., 9., 3., 7., 5.],
       [0., 8., 7., 8., 9., 7., 4., 7.]])]

np.hsplit(a, (3, 4)) 은 어레이 a를 세번째 열 이후, 그리고 네번째 열 이후에서 쪼갠 세개의 어레이를 갖는 어레이를 반환합니다.

관련 페이지

• numpy.floor

이전글/다음글

이전글 : 어레이 합치기

다음글 : NumPy 난수 생성 (Random 모듈)

공유하기