python - numpy.argmax の結果から numpy 配列スライスを抽出する

Question

(3,3) numpy 配列があり、絶対値が最大である要素の符号を把握したいと考えています。

X = [[-2.1,  2,  3],
     [ 1, -6.1,  5],
     [ 0,  1,  1]]

s = numpy.argmax(numpy.abs(X),axis=0) 

必要な要素のインデックスを教えてくれますs = [ 0,1,1].

[ -2.1, -6.1, 5]この配列を使用して要素を抽出し、その符号を把握するにはどうすればよいですか?

Answer 1

これを試して：

# You might need to do this to get X as an ndarray (for example if X is a list)
X = numpy.asarray(X)

# Then you can simply do
X[s, [0, 1, 2]]

# Or more generally
X_argmax = X[s, numpy.arange(X.shape[1])]

Answer 2

部分的な回答:signまたはを使用しsignbitます。

In [8]: x = numpy.array([-2.1, -6.1, 5])

In [9]: numpy.sign(x)
Out[9]: array([-1., -1.,  1.])

In [10]: numpy.signbit(x)
Out[10]: array([ True,  True, False], dtype=bool)