インデックスのリストから配列とリストのサブセットを取得

Question

私はnumpyの配列Xと、要素が対応するpythonリストyを持っています。 I私は部分的なリストは0,1とX=[[1,2],[3,4],[5,6]]とy=[1,2,1]が含まれている場合、例えば、別のアレイrefinedXとrefinedY.

を形成するために使用したいインデックスのリストの一部を持って、私はrefiendX = [[1,2], [3,4]]とrefinedY = [1,2].

どのようにすることができますしたいですこれはPythonで行うのですか？

注：実際の変数Xは（2000,5000）の行列で、部分リストには約500個のアイテムが含まれます。

Answer 1

私は仕事をするいくつかのコードを書いています。あなたが定期的にインデックスを更新する場合は、私が代わりにこれを使用することをお勧めしたい、その場合のように、

X=[[1,2],[3,4],[5,6]] 
Y=[1,2,1] 
indexes = [1,2] 
refinedX = [X[i-1] for i in indexes] 
refinedY = [Y[i-1] for i in indexes] 

これは重いリソースすることができる。

X[indexes[i]-1] #instead of refinedX[i] 

をあなたが特定の値を呼び出すたびに、。