다음은 reshape() 함수를 사용하여 배열의 차원을 바꾸는 예제입니다.

import numpy as np

arr = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9, 10])

new_arr = arr.reshape(2, 5)  # 2행 5열의 배열로 차원 변경

print(new_arr)

위의 예제에서는 arr 배열을 reshape() 함수를 사용하여 2행 5열의 배열로 차원을 변경하였습니다. 결과는 다음과 같습니다.

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

reshape() 함수를 사용할 때 주의해야 할 점은 원래 배열의 요소 개수와 변경하고자 하는 차원의 크기가 맞아야 한다는 것입니다. 예를 들어, 위의 예제에서 arr 배열의 요소 개수가 10개이므로, 변경하고자 하는 차원의 크기도 10개여야 합니다.

이러한 방법으로 NumPy에서 배열의 차원을 바꿀 수 있습니다. NumPy는 다양한 배열 조작 기능을 제공하기 때문에, 배열의 차원을 바꾸는 것 외에도 다른 배열 관련 작업에 유용하게 사용할 수 있습니다.

참고문헌:

• NumPy Reshape documentation: https://numpy.org/doc/stable/reference/generated/numpy.reshape.html