np.where 多个并列条件
时间: 2023-05-09 21:00:57 浏览: 321
np.where详解.ipynb
np.where函数可以在满足特定条件的情况下,给定两个不同值中的任意一个,创建新的数组。
当需要在多个条件下进行选择时,可以使用多个numpy数组,它们的元素对应每个条件。例如,如果有一个名为A的数组,需要在它大于10且小于20的元素中进行选择,并将它们替换为1,其他元素替换为0,则可以按如下方式编写代码:
```
import numpy as np
A = np.random.randint(0, 25, size=(5,5))
B = np.where((A > 10) & (A < 20), 1, 0)
```
可以使用&(与)操作符将两个条件组合起来。此时,只有A数组中同时满足两个条件的元素将被选中。如果条件是或,则可以使用|(或)操作符。与和或操作符可以协同使用来创建更复杂的条件。
总之,np.where函数的多个并列条件,需要使用逻辑运算符来组合条件,以在满足所有条件的情况下选择数组元素,在满足条件时将其替换为给定的值。
阅读全文