提取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)