多个条件where()方法来选择Numpy数组|极客教程

多个条件where()方法来选择Numpy数组

参考：Numpy where function with two conditions

在使用Numpy进行数组操作时，经常会遇到需要根据多个条件来选择数据的情况。Numpy中的where()函数可以帮助我们实现这一目的。本文将详细介绍如何利用where()函数来同时满足两个条件的情况。

基本使用方法

首先，让我们导入Numpy库并创建一个示例数组：

import numpy as np

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

假设我们需要找到数组中同时大于4且为偶数的元素。我们可以通过如下方式使用where()函数来实现：

import numpy as np

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

condition = (array > 4) & (array % 2 == 0)
result = np.where(condition)

print(result)

运行以上代码，将会得到满足条件的元素的索引值：

多个条件where()方法来选择Numpy数组

这表示满足条件的元素分别在数组的第1行第2列和第2行第1列。

除了返回索引值外，我们还可以直接获取满足条件的元素值：

import numpy as np

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

condition = (array > 4) & (array % 2 == 0)
result = np.where(condition)
result_values = array[result]

print(result_values)