NumPy 배열에서 요소의 첫 번째 인덱스를 어떻게 찾을 수 있나요?

NumPy 배열에서 항목의 첫 번째 인덱스 찾기

Python 목록에서 index() 메서드를 사용하면 첫 번째 인덱스를 검색할 수 있습니다. 요소의 발생. NumPy는 배열에 유사한 기능을 제공할 수 있습니까?

답변

예, NumPy는 배열에서 항목의 첫 번째 인덱스를 찾는 편리한 방법을 제공합니다. 다음과 같이 np.where 함수를 활용할 수 있습니다.

import numpy as np

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

itemindex = np.where(array == item)

np.where 함수는 항목이 발견된 행 및 열 인덱스가 포함된 튜플을 반환합니다.

예

배열이 2차원이고 항목이 두 위치에 포함된 경우 then:

array[itemindex[0][0]][itemindex[1][0]]  # equals item
array[itemindex[0][1]][itemindex[1][1]]  # also equals item

이렇게 하면 처음 나타나는 요소를 기준으로 원하는 요소에 쉽게 액세스할 수 있습니다.

위 내용은 NumPy 배열에서 요소의 첫 번째 인덱스를 어떻게 찾을 수 있나요?의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!