如何用numpy.where函数定位数组中特定元素的位置？

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

目录+----------------------------------------+| numpy.where 查找元素位置 |+----------------------------------------+内容：+----------------------------------------+

1.numpy.where() 函数简介

2.二维数组的例子

3.Python numpy.where() 函数示例

4.总结

+----------------------------------------+

1. numpy.where() 函数简介numpy.where() 函数用于查找数组中满足条件的元素的位置。它可以返回满足条件的元素的索引。

2. 二维数组的例子pythonimport numpy as np

arr=np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])print(np.where(arr > 5))

3. Python numpy.where() 函数示例pythonimport numpy as np

arr=np.array([1, 2, 3, 4, 5])print(np.where(arr==3))

4. 总结numpy.where() 函数是查找数组中元素位置的有力工具，可以用于一维和二维数组。