np.where()
时间: 2023-11-20 20:57:41 浏览: 29
`np.where()`是一个NumPy库中的函数,用于根据给定的条件返回输入数组中的元素。它有两种用法:
1. `np.where(condition, x, y)`:当满足条件时输出x,不满足条件则输出y。
举个例子:
```python
import numpy as np
a = np.arange(10)
print(np.where(a > 5, 1, -1))
# 输出:array([-1, -1, -1, -1, -1, -1, 1, 1, 1, 1])
```
2. `np.where(condition)`:只有条件(condition)为True时才返回非零元素的索引。
举个例子:
```python
import numpy as np
a = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
print(np.where(a > 5))
# 输出:(array([1, 2, 2, 2]), array([2, 0, 1, 2]))
```
上面的代码中,输出的是满足条件a>5的元素的索引。
相关问题
np.where
`np.where()` 是 NumPy 库中的一个函数,用于根据给定的条件从数组中选择元素。它的语法如下:
```
np.where(condition[, x, y])
```
其中,`condition` 是一个布尔型数组,表示需要进行选择的元素的条件;`x` 和 `y` 是可选参数,表示在满足条件和不满足条件时的选择。如果只给出 `condition` 参数,则返回满足条件的元素的索引;如果同时给出 `x` 和 `y` 参数,则返回一个数组,其中满足条件的元素用 `x` 参数对应的值替换,不满足条件的元素用 `y` 参数对应的值替换。
np . where
np.where可以使用多个条件来筛选数据,可以通过逻辑运算符(&、|、~)将多个条件组合起来,例如:
np.where((data > ) & (data < 10), 1, )
这个语句的意思是,如果data数组中的元素大于且小于10,则返回1,否则返回。其中&表示逻辑与,|表示逻辑或,~表示逻辑非。可以根据实际需要组合不同的条件来筛选数据。