翻阅Python数据处理入门书籍,其中介绍到了Numpy库的nonzero()函数,平台:Ipython
例码如下:
import numpy as np
x = np.array([[1, 2], [np.nan, 3], [np.nan, np.nan]])
y = x.nonzero()
print(y)
输出结果如下:
(array([0, 0, 1, 1, 2, 2], dtype = int32), array([0, 1, 0, 1, 0, 1], dtype = int32))
在Ipython中 ? np.nonzero() 得到函数的帮助文档:
Signature:np.nonzero(a)
Docstring:
Return the indices of the elements that are non-zero.
returns a tuple of arrays,one for each dimension of 'a',containing the indices of the non-zero elements in that dimension. The values in `a` are always tested and returned in
row-major, C-style order. The corresponding non-zero
values can be obtained with::
a[nonzero(a)]
To group the indices by element, rather than dimension, use::
transpose(