python numpy中nonzero()的用法

72 阅读 0 评论 48 点赞

我是靠谱客的博主感动西牛，最近开发中收集的这篇文章主要介绍python numpy中nonzero()的用法，觉得挺不错的，现在分享给大家，希望可以做个参考。

概述

当使用布尔数组直接作为下标对象或者元组下标对象中有布尔数组时，都相当于用nonzero()将布尔数组转换成一组整数数组，然后使用整数数组进行下标运算。

nonzeros(a)返回数组a中值不为零的元素的下标，它的返回值是一个长度为a.ndim(数组a的轴数)的元组，元组的每个元素都是一个整数数组，其值为非零元素的下标在对应轴上的值。例如对于一维布尔数组b1，nonzero(b1)所得到的是一个长度为1的元组，它表示b1[0]和b1[2]的值不为0(False)。

>>> b1 = np.array([True, False, True, False])
>>> np.nonzero(b1)

(array([0, 2]),)

对于二维数组b2，nonzero(b2)所得到的是一个长度为2的元组。它的第0个元素是数组a中值不为0的元素的第0轴的下标，第1个元素则是第1轴的下标，因此从下面的结果可知b2[0,0]、b[0,2]和b2[1,0]的值不为0：

>>> b2 = np.array([[True, False, True], [True, False, False]])
>>> np.nonzero(b2)

(array([0, 0, 1]), array([0, 2, 0]))

当布尔数组直接做为下标时，相当于使用由nonzero()转换之后的元组作为下标对象：

>>> a = np.arange(3*4*5).reshape(3,4,5)
>>> a[b2]
array([[ 0,
1,
2,
3,
4],

[10, 11, 12, 13, 14],

[20, 21, 22, 23, 24]])
>>> a[np.nonzero(b2)]
array([[ 0,
1,
2,
3,
4],

[10, 11, 12, 13, 14],

[20, 21, 22, 23, 24]])

当下标对象是元组，并且其中有布尔数组时，相当于将布尔数组展开为由nonzeros()转换之后的各个整数数组：

>>> a[1:3, b2]
array([[20, 22, 25],

[40, 42, 45]])
>>> a[1:3, np.nonzero(b2)[0], np.nonzero(b2)[1]]
array([[20, 22, 25],

[40, 42, 45]])

最后

以上就是感动西牛为你收集整理的python numpy中nonzero()的用法的全部内容，希望文章能够帮你解决python numpy中nonzero()的用法所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。

本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。

本文分类：python
浏览次数：72 次浏览
发布日期：2023-11-01 16:10:42
本文链接：https://www.kaopuke.com/article/k-p-k_13_u_23_o_18_f2_12__23__18_3.html

python numpy中nonzero()的用法

概述

最后

评论列表共有 0 条评论

发表评论取消回复

python numpy中nonzero()的用法

概述

最后

相关文章

评论列表共有 0 条评论

发表评论 取消回复

发表评论取消回复