Numpy index with array

https://numpy.org/doc/stable/user/basics.indexing.html

Indexing — NumPy v1.21 Manual

되게 요상하게도 index이 가능하다.

 

예를 들어 x의 행별로 특정 index를 뽑고 싶을 때

각 행에서 뽑을 인덱스를 저장해둔 배열을 y라고 잡는다.

 

그리고 x[np.arange(x.shape[0]), y)라고 하면, 뽑을 인덱스만을 추출하는 것이 가능하다.

 

잘 생각해보면 np.arange(x.shape[0])의 의미가 x의 행의 수만큼의 정수 numpy 배열을 생성하는 것이므로

index에 들어가는 것이 2개의 numpy 배열이다.

 

각각 하나씩 1대1 대응되면서 indexing처리된다고 이해하면 될 듯 싶다.

 

추가적으로 numpy 배열의 경우, 기존의 c++나 python list와 다른 측면이 많으므로

직관적으로 이해하는 것이 필요해보인다.