1D NumPy 배열 전치: 놀랍도록 간단한 솔루션
많은 프로그래머가 1D NumPy 배열을 전치하려고 할 때 혼란을 겪습니다. 문제를 자세히 살펴보고 놀랍도록 간단한 해결책을 찾아보겠습니다.
NumPy에서 .T로 표시된 전치 연산은 배열의 크기를 교환합니다. 그러나 배열이 1D인 경우 전치 후에도 모양이 동일하게 유지됩니다. 결과적으로 2D 배열을 기대하는 사람들에게는 이것이 당황스러울 수 있습니다.
설명하기 위해 다음 코드를 고려하십시오.
import numpy as np a = np.array([5, 4]) print(a) print(a.T)
출력은 다음과 같습니다.
[5 4] [5 4]
보시다시피 1차원 배열 a를 전치해도 모양이 변하지 않습니다. 이는 1D 배열의 전치도 여전히 1D 배열이기 때문입니다.
해결 방법은 1D 배열을 전치하기 전에 2D 배열로 변환하는 것입니다. 이는 본질적으로 배열에 새 축을 삽입하는 np.newaxis 함수를 사용하여 달성할 수 있습니다.
a = np.array([5, 4])[np.newaxis] print(a) print(a.T)
이제 출력은 다음과 같습니다.
[[5 4]] [[5] [4]]
1D 배열 a는 다음과 같습니다. 2D 배열로 성공적으로 변환되었으며 이를 전치하면 원하는 결과가 생성됩니다.
대부분의 경우 다음과 같다는 점에 유의하는 것이 중요합니다. 전치를 위해 1D 배열을 2D 배열로 수동으로 변환할 필요가 없습니다. NumPy는 다양한 작업을 수행할 때 자동으로 1D 배열을 브로드캐스트하므로 차원에 대해 걱정할 필요가 없습니다.
위 내용은 1D NumPy 배열을 전치해도 모양이 바뀌지 않는 이유는 무엇입니까?의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!