2次元配列における np.where での index 取得について

Question

numpy.where を用いて 2次元配列の index を取得することを試みています。
出力される形状を
(array[y0,y1,...], array[x0,x1, ...])
ではなく、
(array[x0,y0], array[x1,y1], ...)
という形状で取得したいです。[x,y]の組で，かつ，全体がタプルに囲まれている形が理想です。

適切な変換方法があれば教えていただきたいです。
よろしくお願いいたします。

サンプルコード

import numpy as np

A = np.random.randn(3,3)
indices = np.where(A > 0)
print(indices)

出力

(array([1, 1, 1]), array([0, 1, 2]))

#理想
(array([1,0],array[1,1],array[1,2])

Answer 1

import numpy as np

A = np.random.randn(3,3)
indices = tuple(np.argwhere(A.T > 0))

print(A)
print(indices)

#
[[ 1.39352534 -2.00166684  0.74056198]
 [ 0.84778677  1.00009956  1.59520949]
 [-0.44935818 -0.43921361 -0.42835765]]
(array([0, 0]), array([0, 1]), array([1, 1]), array([2, 0]), array([2, 1]))

Answer 2

以下です。

>>> print(indices)
(array([0, 0, 2], dtype=int64), array([1, 2, 1], dtype=int64))
>>> print(tuple(np.vsplit(np.stack(indices[::-1]).T, indices[0].shape[0])))
(array([[1, 0]], dtype=int64), array([[2, 0]], dtype=int64), array([[1, 2]], dtype=int64))

Comments

[majita]

助かりました。迅速な回答ありがとうございました。