python中的numpy.where()函数是一个条件函数,用于返回满足指定条件的元素的索引。它的工作原理是基于给定条件的布尔数组来选择元素。如果条件为真,则返回对应元素的索引;如果条件为假,则返回默认值或者None。

8 浏览
0 Comments

python中的numpy.where()函数是一个条件函数,用于返回满足指定条件的元素的索引。它的工作原理是基于给定条件的布尔数组来选择元素。如果条件为真,则返回对应元素的索引;如果条件为假,则返回默认值或者None。

我正在使用numpy并查阅文档,偶然发现了一些神奇之处。具体来说,我在谈论numpy.where()

>>> x = np.arange(9.).reshape(3, 3)
>>> np.where( x > 5 )
(array([2, 2, 2]), array([0, 1, 2]))

他们是如何实现可以将类似x > 5这样的内容传递给一个方法的内部呢?我猜这与__gt__有关,但我希望得到一个详细的解释。

0