Python中where函数怎样定义和使用-群英

上一篇：Python命令行库click能做什么，用法是怎样的下一篇：基于Pygame怎样实现吃水果的小游戏

今天这篇我们来学习和了解“Python中where函数怎样定义和使用”，下文的讲解详细，步骤过程清晰，对大家进一步学习和理解“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函数怎样定义和使用”的文章就介绍到这了,更多相关的内容，欢迎关注群英网络，小编将为大家输出更多高质量的实用文章！

标签： where函数

免责声明：本站发布的内容（图片、视频和文字）以原创、转载和分享为主，文章观点不代表本网站立场，如果涉及侵权请联系站长邮箱：mmqy2019@163.com进行举报，并提供相关证据，查实之后，将立刻删除涉嫌侵权内容。

上一篇：Python命令行库click能做什么，用法是怎样的下一篇：基于Pygame怎样实现吃水果的小游戏