极客笔记Numpy where函数
numpy.where() 是一个重要的Numpy函数，用于根据给定的条件返回输入数组中满足条件的元素的索引。
numpy.where()语法说明
该函数的语法如下：
numpy.where(condition[, x, y])
参数说明：
– condition: 数组或条件表达式，指定筛选条件
– x：可选参数，满足条件返回x，默认情况下返回满足条件的元素索引
– y：可选参数，不满足条件返回y
示例
让我们通过几个示例来详细了解numpy.where()函数的用法。
示例 1：返回满足条件的元素索引
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
indices = np.where(arr > 2)
print(indices)
输出结果：
示例 2：返回满足条件的元素
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
new_arr = np.where(arr > 2, arr, 0)
print(new_arr)
输出结果：
示例 3：测试数组中的奇偶性
后使用np.where函数根据条件condition，将arr中符合条件的奇偶元素替换为相应的元素。下面是一个示例代码：
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
condition = np.array([True, False, True, False, True])
result = np.where(condition, arr, 0)
print(result)
运行结果：
在这个示例中，我们首先创建了一个数组arr，然后创建了一个条件数组condition，表示对应位置的元素是否符合条件。最后使用np.where函数根据条件condition，将arr中符合条件的元素替换为对应位置的元素，不符合条件的元素替换为0。
总结
通过以上示例，我们可以看到numpy.where()函数的用法非常灵活，可以根据实际需求返回满足条件的元素索引或元素本身。在实际应用中，我们可以利用numpy.where()函数快速处理数组数据，提高数据处理的效率。