2016-10-06 8 views
1

私が持っているとしてアレイ:INDの要素はAの行数を表す引張要素は、インデックス配列に基づいて

ind = np.asarray([1,2,0,2,2]) 

:私としてもインデックス配列を有する

import numpy as np 
A = np.arange(15).reshape(3, 5) 

すなわちA.

の各列のために私は、列0からind[0] = 1要素を引き出したいです私は

所望の出力のカラム4からind[4] = 2要素をプルする である:

Numpy's fancy-indexing使用
5, 11, 2, 13, 14 
+1

'A = [A [IND [x]は、xと] for x in range(5)] ' – polka

答えて

2

-

A[ind,np.arange(ind.size)] 
関連する問題