アウト変数の値ごとにパンダシリーズ適用関数

Question

私は各パンダシリーズから選択されたパンダシリーズとインデックスアレイを持っています。

これはおもちゃのデータです。

index = np.array([0,1,0]) 
row = pd.Series([np.array([1,2,3]),np.array([2,3,4]),np.array([3,4,5])]) 

おもちゃのデータの上から以下のようなデータを処理したい。

[row.iloc[0][0:0+2], 
row.iloc[1][1:1+2], 
row.iloc[2][0:0+2]] 

[array([1, 2]), array([3, 4]), array([3, 4])] 

私はそれが動作しない、

row.apply(lambda x, index: x[index:index+2]) 

のような機能を適用する使用しようとしたが。

これを効率的に処理する方法はありますか？

私は膨大なデータを持っているので、効率的な方法が必要です。

Answer 1

zipをプレーンリストの理解度とともに使用できます。

[r[i:i+2] for i, r in zip(index, row)] 
Out[127]: 
[[1, 2], [3, 4], [3, 4]]