这篇“Python where函数怎么使用”文章的知识点大部分人都不太理解,所以小编给大家总结了以下内容,内容详细,步骤清晰,具有一定的借鉴价值,希望大家阅读完这篇文章能有所收获,下面我们一起来看看这篇“Python where函数怎么使用”文章吧。
where函数是numpy库中的,通常需要先加载numpy库,再调用该函数。函数的基本调用语法有两种,一种是:
import numpy as np np.where(arry)
此时,np.where函数输出arry中“真”值的坐标(‘真’也可以理解为非0)。或者说np.where函数从arry中返回满足特定条件的元素。比如,它会返回满足特定条件数值的索引位置。
另一种是:
import numpy as np np.where(cond, x, y)
此时,np.where函数满足cond条件输出x,不满足输出y。为了让大家对where函数定义有更清晰的理解,接下来以具体实例进行阐述,方便大家理解记忆。
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),)
此时,np.where函数返回数值大于5的索引位置。
y = np.array(range(1, 10)) print(y) print(np.where(y>5, 'm_5', 'lq_5'))
得到结果:
[1 2 3 4 5 6 7 8 9]
['lq_5' 'lq_5' 'lq_5' 'lq_5' 'lq_5' 'm_5' 'm_5' 'm_5' 'm_5']
y是一个初始值为1,终值为9,步长为1的等差数列。此时,np.where函数满足y>5输出’m_5’,不满足输出’lq_5’。
print(np.arange(10)) print(np.where(np.arange(10)<5, '吃苹果', '吃榴莲'))
得到结果:
[0 1 2 3 4 5 6 7 8 9]
[‘吃苹果’ ‘吃苹果’ ‘吃苹果’ ‘吃苹果’ ‘吃苹果’ ‘吃榴莲’ ‘吃榴莲’ ‘吃榴莲’ ‘吃榴莲’ ‘吃榴莲’]
此时,np.where函数满足np.arange(10)中数值小于5输出’吃苹果’,不满足输出’吃榴莲’。
y = np.array(range(1, 10)) print(y) print(np.where(np.mod(y, 2)==0, '2b', 'n_2b'))
得到结果:
[1 2 3 4 5 6 7 8 9]
['n_2b' '2b' 'n_2b' '2b' 'n_2b' '2b' 'n_2b' '2b' 'n_2b']
此时,np.where函数满足y中数值除以2余数为0输出’2b’,不满足输出’n_2b’。
x = np.array([[0, 1, 2], [3, 0, 0], [6, 0, 8]]) print(x[np.where(x)])
得到结果:
[1 2 3 6 8]
此时,np.where函数取出x中所有非0数,生成一个新的数列。
以上就是关于“Python where函数怎么使用”这篇文章的内容,相信大家都有了一定的了解,希望小编分享的内容对大家有帮助,若想了解更多相关的知识内容,请关注亿速云行业资讯频道。
免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。