3

Numpy を使用して、b と c を指定して、以下の結果を達成したいと思います。スタッキング関数を調べましたが、機能させることができません。誰か助けてくれませんか？

import numpy as np

a=range(35,135)

b=np.reshape(a,(10,10))
c=np.array([[5,5],[5,6],[5,7],[6,5],[6,6],[6,7],[7,5],[7,6],[7,7]])

結果は次のようになります。

np.array([[5,5,90],[5,6,91],[5,7,92],[6,5,100],[6,6,101],[6,7,102],
          [7,5,110],[7,6,111],[7,7,112]])
4

1 に答える 1

8

ふぅ！これは大当たりでした。まず、numpy のファンシー インデックスを使用して、必要なアイテムを引き出します。

>>> b[tuple(c.T)]
array([ 90,  91,  92, 100, 101, 102, 110, 111, 112])

c次に、残っている唯一のことは、その配列をusingに対してスタックバックすることcolumn_stackです:

>>> np.column_stack((c,b[tuple(c.T)]))
array([[  5,   5,  90],
       [  5,   6,  91],
       [  5,   7,  92],
       [  6,   5, 100],
       [  6,   6, 101],
       [  6,   7, 102],
       [  7,   5, 110],
       [  7,   6, 111],
       [  7,   7, 112]])
于 2013-06-04T13:40:55.780 に答える