如何用numpy实现类似where函数的功能来查找数组中元素的位置？

本文共计1221个文字，预计阅读时间需要5分钟。

目录- 使用numpy.where查找元素位置的例子- 二维数组的例子- Python numpy.where() 函数- 总结- numpy.where(condition, x, y) 详细用法- 官方文档请见：

使用numpy.where查找元素位置的例子

在NumPy中，`numpy.where()` 函数可以用来找出数组中满足特定条件的元素的位置。以下是一些简单的例子：

二维数组的例子

pythonimport numpy as np

创建一个二维数组array_2d=np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

找出数组中值为5的所有元素的索引indices=np.where(array_2d==5)print(indices)

Python numpy.where() 函数

`numpy.where()` 函数的基本使用方法如下：

pythonnumpy.where(condition[, x, y])

其中：- `condition` 是一个布尔数组，用于指定条件。- `x` 是在条件为假时返回的值。- `y` 是在条件为真时返回的值。

总结

`numpy.where()` 是一个非常有用的函数，可以快速找出数组中满足条件的元素的位置。