numpy知识点

Posted 2020-11-03 zhaop

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了numpy知识点相关的知识，希望对你有一定的参考价值。

1、nonzero

对于一维数据来说，将返回符合条件的下标

>>> b1 = np.array([True, False, True, False])
>>> np.nonzero(b1)
    (array([0, 2]),)

对于二维数据来说，将返回两维元组，第一维是符合条件的 x的索引，第二维是符合条件的y的索引

>>> b2 = np.array([[True, False, True], [True, False, False]])
>>> np.nonzero(b2)
    (array([0, 0, 1]), array([0, 2, 0]))

以上是关于numpy知识点的主要内容，如果未能解决你的问题，请参考以下文章