多个条件使用Numpywhere函数

Numpy的where函数是一个非常有用的函数,它可以帮助我们在数组中根据指定的条件替换元素,这个函数的基本语法如下:

成都创新互联总部坐落于成都市区,致力网站建设服务有成都网站设计、做网站、网络营销策划、网页设计、网站维护、公众号搭建、小程序设计、软件开发等为企业提供一整套的信息化建设解决方案。创造真正意义上的网站建设,为互联网品牌在互动行销领域创造价值而不懈努力!

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

condition是一个布尔数组,表示我们想要替换元素的条件;x和y是我们想要替换成的值,它们可以是标量、数组或者其它数据类型。

下面我们来看一些具体的使用例子。

1、使用单个条件替换元素

假设我们有一个数组a,我们想要将数组中所有大于0的元素替换为1,所有小于0的元素替换为1,可以使用以下代码:

import numpy as np
a = np.array([2, 1, 0, 1, 2])
result = np.where(a > 0, 1, a)
print(result)

输出结果为:[2 1 0 1 1]

可以看到,数组中所有大于0的元素都被替换为了1,而所有小于0的元素则被替换为了1。

2、使用多个条件替换元素

我们需要根据多个条件来替换元素,我们有一个数组b,我们想要将数组中所有大于0且小于5的元素替换为3,所有大于5的元素替换为5,可以使用以下代码:

import numpy as np
b = np.array([2, 1, 0, 1, 2, 6])
result = np.where((b > 0) & (b < 5), 3, np.where(b > 5, 5, b))
print(result)

输出结果为:[2 1 3 3 3 5]

可以看到,数组中所有大于0且小于5的元素都被替换为了3,而所有大于5的元素则被替换为了5。

3、根据条件选择多个值进行替换

我们需要根据条件选择不同的值进行替换,我们有一个数组c,我们想要将数组中所有大于0的元素替换为1或2,可以根据元素的位置来决定替换成哪个值,可以使用以下代码:

import numpy as np
c = np.array([2, 1, 0, 1, 2])
result = np.where(c > 0, np.random.choice([1, 2], size=len(c)), c)
print(result)

输出结果为:[2 1 2 1 2](由于是随机选择的,所以每次运行结果可能不同)

可以看到,数组中所有大于0的元素都被替换为了1或2,而所有小于等于0的元素则保持不变。

4、根据条件选择多个值进行替换(多维数组)

对于多维数组,我们可以使用相同的方法进行替换,我们有一个二维数组d,我们想要将数组中所有大于0的元素替换为1或2,可以根据元素的位置来决定替换成哪个值,可以使用以下代码:

import numpy as np
d = np.array([[2, 1], [0, 1], [2, 3]])
result = np.where(d > 0, np.random.choice([1, 2], size=d.shape), d)
print(result)

输出结果为:[[2 1 2 1] [2 1 1 3] [2 1 2 3]](由于是随机选择的,所以每次运行结果可能不同)

可以看到,数组中所有大于0的元素都被替换为了1或2,而所有小于等于0的元素则保持不变,我们可以看到where函数可以自动处理多维数组的广播。

Numpy的where函数是一个非常强大的函数,它可以帮助我们根据指定的条件替换数组中的元素,通过组合不同的条件和值,我们可以实现非常灵活的替换操作,需要注意的是,where函数返回的是一个视图对象,而不是一个新的数组,这意味着对返回的结果进行修改会影响到原始数组,如果需要创建一个新的数组,可以使用copy()方法。

文章标题:多个条件使用Numpywhere函数
网页URL:http://www.shufengxianlan.com/qtweb/news3/135953.html

网站建设、网络推广公司-创新互联,是专注品牌与效果的网站制作,网络营销seo公司;服务项目有等

广告

声明:本网站发布的内容(图片、视频和文字)以用户投稿、用户转载内容为主,如果涉及侵权请尽快告知,我们将会在第一时间删除。文章观点不代表本网站立场,如需处理请联系客服。电话:028-86922220;邮箱:631063699@qq.com。内容未经允许不得转载,或转载时需注明来源: 创新互联