Python中的where函数该怎么使用？Python第三方库numpy函数where语法是什么 _python

python第三方库numpy是用来完成数据生成或者是各种数据处理的，而在这个其中有一个函数where能够实现快捷简单的数据拆解。下文会详细介绍np.where()函数的语法以及其使用方法，想学习和了解的小伙伴就一起往下看看吧。

一、where函数语法
where这个函数的作用就是对一个数组进行判断，然后返回其中符合表达式条件的值。简单来说就是对数组元素进行一次if条件判断，代码示例如下所示：
import numpy as npnp.where(cond, x, y)上面代码示例就是where函数的语法，cond参数为必选项，传入的就是条件判断表达式。而后面两个参数x和y则是可选项，表示符合以及不符合条件两种情况下返回的值。
二、where函数使用
使用之前需要调用array()方法来生成数组对象，因为它第一个参数表达式进行判断的对象就需要是数组类型，代码示例如下所示：
y = np.array([1, 5, 6, 8, 1, 7, 3, 6, 9])print(np.where(y>5))(array([2, 3, 5, 7, 8], dtype=int64),)以上where函数的意思就是，将数组对象y之中值大于5的元素索引返回出来，并且它会将这个数组的元素数据类型一并返回。
另外一种调用方式就是添加x和y这两个可以参数，这两个可选参数的数据类型为object，也就是任意数据类型都可以，详细代码示例如下所示：
y = np.array(range(1, 10))print(np.where(y>5, 'm', 'lq'))['lq' 'lq' 'lq' 'lq' 'lq' 'm' 'm' 'm' 'm']以上代码先使用range()函数生成了1-9数字的可迭代对象并将其转为数组，然后调用where函数执行判断。当数组元素值大于5时就返回m字符串，否则的话就返回lq字符串，有点类似循环调用三元算符。
【Python中的where函数该怎么使用？Python第三方库numpy函数where语法是什么】以上就是关于“Python中的where函数该怎么使用？Python第三方库numpy函数where语法是什么”的全部内容了，希望对你有所帮助。

微信扫码Python课,免费报名

现在报名赠100例知识点合集

Python中的where函数该怎么使用？Python第三方库numpy函数where语法是什么

猜你喜欢