在Numpy中,我们可以使用where()
方法来根据多个条件选择数组元素。where()
方法的语法如下:
numpy.where(condition[, x, y])
参数说明:
condition
:一个布尔数组,用于指定满足条件的索引。
x
:可选参数,当condition
为True时,返回的值。
y
:可选参数,当condition
为False时,返回的值。
下面是一个使用多个条件选择Numpy数组的示例:
1、我们创建一个Numpy数组:
import numpy as np arr = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9])
2、我们定义两个条件:大于5和小于等于5:
greater_than_5 = arr > 5 less_than_or_equal_to_5 = arr <= 5
3、接下来,我们使用where()
方法根据这两个条件选择数组元素:
result = np.where((greater_than_5 & less_than_or_equal_to_5), "A", "B")
这里,我们使用了逻辑与操作符(&)将两个条件组合在一起,当数组元素同时满足大于5和小于等于5的条件时,我们将其替换为"A";否则,我们将其替换为"B"。
4、我们打印结果:
print(result)
输出结果:
['B' 'B' 'B' 'A' 'A' 'A' 'A' 'A']
可以看到,数组中大于5且小于等于5的元素被替换为了"A",其他元素被替换为了"B"。
最新评论
本站CDN与莫名CDN同款、亚太CDN、速度还不错,值得推荐。
感谢推荐我们公司产品、有什么活动会第一时间公布!
我在用这类站群服务器、还可以. 用很多年了。