2013-01-04 4 views
8

여러 인덱스를 사용하여 얻는 방법 :NumPy와 배열의 값이 나는이처럼 보이는 NumPy와 배열이

arr = [100.10, 200.42, 4.14, 89.00, 34.55, 1.12] 

을 어떻게 인덱스이 배열에서 여러 값을 얻을 수 있습니다 : 예를 들어

, 인덱스 위치 1, 4 및 5에서 값을 얻으려면 어떻게해야합니까? 이 같은 시도

arr[1, 4, 5] 
+0

에 규정 np.take을 사용하는 것입니다, 당신이 시도 않았다하면 NumPy와의 다차원 색인 작업을 수행하는 방법이다. – Iguananaut

+1

또한 가치있는 일을 위해서, 당신은 * 질적 인 배열을 가지고 있지 않습니다. 당신은 목록을 가지고 있습니다 (쉽게 numpy 배열로 변환 될 수 있습니다). – mgilson

답변

25

을 :

내가 잘못이다, 이런 식으로 뭔가를 시도하고 있었다

>>> arr = np.array([100.10, 200.42, 4.14, 89.00, 34.55, 1.12]) 
>>> arr[[1,4,5]] 
array([ 200.42, 34.55, 1.12]) 

을 그리고 다차원 배열을 위해 :

>>> arr = np.arange(9).reshape(3,3) 
>>> arr 
array([[0, 1, 2], 
     [3, 4, 5], 
     [6, 7, 8]]) 
>>> arr[[0, 1, 1], [1, 0, 2]] 
array([1, 3, 5]) 
+0

위대한 마음은 비슷하게 생각합니다 : P –

+0

인덱스가 2D 인 경우 어떻게해야합니까? 2D 배열을 인덱스로 사용하여 2D 배열을 인덱싱 할 수 있습니까? – dashesy

4

당신은

가까웠다
>>> print arr[[1,4,5]] 
[ 200.42 34.55 1.12]