Numpy库函数where使用方法有哪些?Python如何使用where函数

【Numpy库函数where使用方法有哪些?Python如何使用where函数】python数据生成和处理第三方库numpy里面有一个方法叫做where() , 它能够来对数组对象进行简单的三元运算判断操作 。下文会以几个示例来详解该怎么where函数的使用方法有哪些 , 感兴趣和想学习的小伙伴就一起往下看吧 。

Numpy库函数where使用方法有哪些?Python如何使用where函数

文章插图
1.where()函数需要使用数组作为参数 , 那么可以直接调用arange()函数来生成数组对象进行判断 。然后设置一个界限值 , 将小于和大于这个界限值的数字使用特定值返回 , 代码示例如下所示:
import numpy as npprint(np.where(np.arange(10)<5, '果', '莲'))以上代码之中会生成从0-9这十个整数的数组 , 然后where()函数会将数组中所有值取出来判断一下它们是否小于5 。小于5的数字就会转为字符串‘果’返回到列表中 , 大于等于5的数字则是转为字符串‘莲’返回到列表中 。
2.除了使用大于小于号之外 , 它还可以使用除法函数加上双等号 , 来判断元素计算之后的结果是否符合某个条件 。只要传进来的参数值是一个if条件判断表达式即可 , 代码如下所示:
y = np.array(range(1, 10))print(np.where(np.mod(y, 2)==0, '2b', 'n_2b'))mod()函数是用来进行取余的 , 如果数组y中的元素除以2等于0 , 那么就返回2b , 否则返回n_2b 。
3.以上操作都是对一维数组进行处理的 , 而where()函数实际上能够对任意维度数组对象完成操作 。那么对二维数组进行操作可以取出非0数据 , 只需要使用数组切片语法将这个函数作为参数传入即可 , 代码示例如下所示:
x = np.array([[0, 1, 2], [3, 0, 0], [6, 0, 8]])print(x[np.where(x)])[1 2 3 6 8]以上就是关于“Numpy库函数where使用方法有哪些?Python如何使用where函数”的全部内容了 , 希望对你有所帮助 。
Numpy库函数where使用方法有哪些?Python如何使用where函数

文章插图
Numpy库函数where使用方法有哪些?Python如何使用where函数

文章插图
Numpy库函数where使用方法有哪些?Python如何使用where函数

文章插图
微信扫码Python课,免费报名
Numpy库函数where使用方法有哪些?Python如何使用where函数

文章插图
现在报名赠100例知识点合集
Numpy库函数where使用方法有哪些?Python如何使用where函数

文章插图

    推荐阅读