如何用numpy查找数组中特定区间元素？
- 内容介绍
- 文章标签
- 相关推荐
本文共计474个文字，预计阅读时间需要2分钟。
使用NumPy的`where`函数，可以像MATLAB的`find`函数一样，返回一个区间内元素的索引位置。以下是一个简化的示例代码，用于在多维度数组中查找特定条件下的元素索引：
pythonimport numpy as np
def find_indices(arr, condition): return np.where(condition(arr))
示例arr=np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])condition=lambda x: x > 5indices=find_indices(arr, condition)print(indices)
这段代码首先定义了一个`find_indices`函数，它接受一个NumPy数组和一个条件函数。然后，使用`np.where`来查找满足条件的元素索引。在示例中，我们查找数组`arr`中所有大于5的元素的位置。
找了半天，以为numpy的where函数像matlab 的find函数一样好用，能够返回一个区间内的元素索引位置。结果没有。。
本文共计474个文字，预计阅读时间需要2分钟。
使用NumPy的`where`函数，可以像MATLAB的`find`函数一样，返回一个区间内元素的索引位置。以下是一个简化的示例代码，用于在多维度数组中查找特定条件下的元素索引：
pythonimport numpy as np
def find_indices(arr, condition): return np.where(condition(arr))
示例arr=np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])condition=lambda x: x > 5indices=find_indices(arr, condition)print(indices)
这段代码首先定义了一个`find_indices`函数，它接受一个NumPy数组和一个条件函数。然后，使用`np.where`来查找满足条件的元素索引。在示例中，我们查找数组`arr`中所有大于5的元素的位置。
找了半天，以为numpy的where函数像matlab 的find函数一样好用，能够返回一个区间内的元素索引位置。结果没有。。