NumPy 다차원 배열의 개별 열에 어떻게 액세스합니까?

NumPy 다차원 배열의 개별 열에 액세스

NumPy는 다차원 배열 작업을 위한 강력한 도구를 제공합니다. 이러한 배열의 개별 열에 액세스해야 하는 경우가 많습니다. 이 가이드에서는 이를 쉽게 달성하는 방법을 보여줍니다.

다음 NumPy 배열을 상상해 보세요.

<code class="python">test = np.array([[1, 2], [3, 4], [5, 6]])</code>

이 배열의 i번째 행에 액세스하려면 test[i]를 사용하면 됩니다. 그러나 i번째 열에 액세스해야 하는 경우 구문이 약간 다릅니다.

i번째 열에 액세스하려면 다음 구문을 사용합니다.

<code class="python">test[:, i]</code>

예를 들어 첫 번째 열에 액세스하려면 열:

<code class="python">>>> test[:, 0]
array([1, 3, 5])</code>

첫 번째 인덱스(:)는 모든 행을 선택하고 두 번째 인덱스(0)는 열을 지정합니다.

이 기능은 NumPy의 섹션 1.4에 설명되어 있습니다. 참조. 성능 측면에서 이러한 방식으로 열에 액세스하는 것은 일반적으로 루프 내에서 각 요소에 개별적으로 액세스하는 것보다 효율적이고 훨씬 빠릅니다.

위 내용은 NumPy 다차원 배열의 개별 열에 어떻게 액세스합니까?의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!