numpy中关于where函数的用法

更新时间：2024年02月02日 14:50:47 作者：坝坝头伯爵

这篇文章主要介绍了numpy中关于where函数的用法,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教

关于where函数用法

在我的理解中np.where函数有三个用法

1. np.where()[0] 和 np.where()[1]

where在我的理解中是一个寻找数组中某个元素的函数

在此用法中np.where()[0] 表示行索引，np.where()[1]表示列索引

具体如下：

import numpy as np
array = np.arange(12).reshape(3,4)
print('array:', array)
print('np.where(array > 5):', np.where(array > 5))
print('array[np.where(a > 5)]:', array[np.where(array > 5)])
print('np.where(array > 5)[0]:', np.where(array > 5)[0])
print('np.where(array > 5)[1]:', np.where(array > 5)[1])
print(array[np.where(array > 5)[0], np.where(array > 5)[1]])

array: [[ 0 1 2 3]
 [ 4 5 6 7]
 [ 8 9 10 11]]
np.where(array > 5): (array([1, 1, 2, 2, 2, 2]), array([2, 3, 0, 1, 2, 3]))
array[np.where(array > 5)]: [ 6 7 8 9 10 11]
np.where(array > 5)[0]: [1 1 2 2 2 2]
np.where(array > 5)[1]: [2 3 0 1 2 3]
[ 6 7 8 9 10 11]

2. np.where(condition, x, y)

满足condition 这个条件的输出x，不满足输出y。

如下所示：

小于2的数改为0，否则改为10

import numpy as np
t2=np.array([[1,2,3],[1,2,3]])
print(np.where(t2<2,0,10))

[[ 0 10 10]
 [ 0 10 10]]#输出

3. np.where(condition)

只有条件 (condition)，没有x和y，则输出满足条件 (即非0) 元素的坐标 (类似于numpy.nonzero)。

举个简单的例子：

import numpy as np
t2=np.array([[1,2,3],[1,2,3]])
print(np.where(t2<2))

(array([0, 1], dtype=int64), array([0, 0], dtype=int64)) #输出

所以 np.where会输出每个符合要求的元素的对应的坐标

需要注意的一点是：

输入的不能直接是list，需要转为array或者为array才行。

比如range(10)和np.arange(10)后者返回的是数组，使用np.where才能达到效果。

总结

以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

python中return的返回和执行实例
今天小编就为大家分享一篇python中return的返回和执行实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2019-12-12
Python3.6安装卸载、执行命令、执行py文件的方法详解
这篇文章主要介绍了Python3.6之安装卸载、执行命令、执行py文件的方法，本文给大家介绍的非常详细，具有一定的参考借鉴价值,需要的朋友可以参考下
2020-02-02
python使用requests设置读取超时时间
在Python中,使用requests库进行网络请求时,可以通过设置 timeout参数来指定读取超时时间,本文就来介绍一下,具有一定的参考价值,感兴趣的可以了解一下
2023-11-11
Python中三个不可思议的返回功能分享
这篇文章主要给大家介绍了Python中三个不可思议的返回功能,文中通过实例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
2022-01-01
python实现自动打卡小程序
这篇文章主要为大家详细介绍了python实现自动打卡小程序，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2021-03-03
Python 字符串中的字符倒转
实现字符倒转的python函数
2008-09-09
Python 带你快速上手 Apache APISIX 插件开发
Apache APISIX Python Runner 来了，社区中的小伙伴们在开发 Apache APISIX 插件时又多了一种新选择，本文将用实列向大家介绍,需要的朋友可以参考下面文章内容
2021-09-09
matplotlib调整子图间距,调整整体空白的方法
今天小编就为大家分享一篇matplotlib调整子图间距,调整整体空白的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2018-08-08
Python基于pandas实现json格式转换成dataframe的方法
这篇文章主要介绍了Python基于pandas实现json格式转换成dataframe的方法,结合实例形式分析了Python使用pandas模块操作json数据转换成dataframe的相关操作技巧与注意事项,需要的朋友可以参考下
2018-06-06
python实现循环语句1到100累和
这篇文章主要介绍了python循环语句1到100累和方式，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教
2022-05-05