np.where()函数

np.where()函数是一个基于条件表达式返回输入数组中元素的索引的函数。它接收一个条件表达式和两个可选参数x和y，返回一个数组，其中包含元素的索引，这些元素在输入数组中满足条件表达式。

当只提供条件表达式时，np.where()函数返回一个包含满足条件的元素索引的元组。当提供x和y参数时，它会根据条件表达式返回x或y中的相应元素的索引。

以下是一个示例，演示了如何使用np.where()函数：

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
cond = np.array([True, False, True, False, False])

# 使用条件表达式返回元素的索引
result = np.where(cond)
print(result) # 输出：(array([0, 2]),)

# 使用x和y参数返回相应的元素的索引
result = np.where(cond, arr, -1)
print(result) # 输出：[1, -1, 3, -1, -1]

在上面的示例中，我们首先定义了一个输入数组和一个布尔条件数组。我们使用np.where()函数来查找满足条件的元素的索引。在第一个示例中，我们只提供了条件表达式，因此函数返回一个包含满足条件的元素索引的元组。在第二个示例中，我们使用x和y参数来返回相应的元素的索引。如果条件为True，则返回输入数组中相应位置的元素，否则返回-1。

相关问题

np.where函数

np.where函数是numpy库中的一个非常实用的函数，主要有两个用法。第一个用法是np.where(condition, x, y)，它会返回满足condition的索引。第二个用法是首先传入y，然后如果满足条件condition，那么y中原来的数值变为x，否则不变化。相当于[xv if c else yv for (c,xv,yv) in zip(condition,x,y)]。这个函数可以帮助我们快速地对数组进行筛选和替换操作。

python中np.where函数

np.where函数是numpy库中的一个函数，它的作用是根据指定的条件，返回符合条件的元素的索引或值。np.where函数的语法如下：

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

其中，condition是指定的条件，x和y是可选的参数，用于指定符合条件和不符合条件的返回值。

当只有condition参数时，np.where函数返回符合条件的元素的索引。

当有x和y参数时，np.where函数会根据condition的值，返回符合条件的元素的值或不符合条件的元素的值。

例如，以下代码会将数组a中小于0的元素替换为0，其他元素不变：

import numpy as np

a = np.array([-1, 2, -3, 4, 5])

b = np.where(a < 0, 0, a)

print(b)

输出结果为：

[0 2 0 4 5]

其中，np.where(a < 0, 0, a)的作用是将数组a中小于0的元素替换为0，其他元素不变。