NumPy 배열에 차원 추가

1. numpy.expand_dims()함수를 사용하여 NumPy 배열에 차원 추가
2. Python에서numpy.newaxis함수를 사용하여 NumPy 배열에 차원 추가

이 자습서에서는 Python에서 NumPy 배열에 새 차원을 추가하는 방법을 소개합니다.

numpy.expand_dims()함수를 사용하여 NumPy 배열에 차원 추가

numpy.expand_dims()함수는 NumPy 배열에 새로운 차원을 추가합니다. 확장 할 배열과 새 축을 인수로 사용하고 추가 차원이있는 새 배열을 반환합니다. numpy.expand_dims()함수의axis매개 변수 내에서 확장 할 축을 지정할 수 있습니다. 다음 코드 예제를 참조하십시오.

import numpy as np

array = np.array([1, 2, 3])
print(array.shape)

array = np.expand_dims(array, axis=0)
print(array.shape)

array = np.append(array, [[4, 5, 6]], axis=0)
print(array)

출력:

(3,)
(1, 3)
[[1 2 3]
 [4 5 6]]

위 코드에서 먼저np.array()함수를 사용하여 1D 배열array를 생성하고array.shape속성을 사용하여array의 모양을 인쇄했습니다. 그런 다음np.expand_dims(array, axis=0)함수를 사용하여array를 2D 배열로 변환하고array.shape속성을 사용하여array의 새 모양을 인쇄했습니다. 마지막으로np.append()함수를 사용하여array에 새 요소를 추가하고array의 요소를 인쇄했습니다.

Python에서numpy.newaxis함수를 사용하여 NumPy 배열에 차원 추가

이전 접근 방식이 작업을 수행하고 현재로서는 잘 작동합니다. 유일한 문제는 이전 방법이 더 이상 사용되지 않으며 향후 Python의 최신 버전에서 작동하지 않을 것이라는 점입니다. numpy.newaxis메소드는 이전 메소드와 동일한 목표를 달성하는 데에도 사용할 수 있지만 코드와 복잡성이 훨씬 적습니다. 이 방법을 사용하면 Python의 이후 버전에서 지원되지 않는 것에 대해 걱정할 필요가 없습니다. numpy.newaxis메소드는 Python의 배열에 새로운 차원을 추가합니다.

import numpy as np

array = np.array([1, 2, 3])
print(array.shape)

array = array[np.newaxis]
print(array.shape)

array = np.append(array, [[4, 5, 6]], axis=0)
print(array)

출력:

(3,)
(1, 3)
[[1 2 3]
 [4 5 6]]

array[np.newaxis]메소드를 사용하여array를 2D 배열로 변환하고array.shape속성을 사용하여array의 새 모양을 인쇄했습니다. 마지막으로np.append()함수를 사용하여array에 새 요소를 추가하고array의 요소를 인쇄했습니다.