numpyのインデックスについて

Question

pythonを使っています。numpyで、ある行列の各行に対し、インデックス用の配列が指定した位置に所定の数値を入れたいと思っています。

例えば、行列がゼロで初期化された4x3の大きさとして、インデックス用の配列が[0,2,1,0]のとき、
[[1,0,0]
[0,0,1]
[0,1,0]
[1,0,0]]
という結果が得られるようにしたいです。

どのように書けばよいでしょうか。

Answer 1

自己解決しました。

x = np.zeros((4,3))
y = np.array([0,2,1,0])
x[xrange(len(y)), y] = 1