Python中如何使用numpy.where()函数

小编给大家分享一下Python中如何使用numpy.where()函数,希望大家阅读完这篇文章后大所收获,下面让我们一起去探讨吧!

创新互联公司-专业网站定制、快速模板网站建设、高性价比抚州网站开发、企业建站全套包干低至880元,成熟完善的模板库,直接使用。一站式抚州网站制作公司更省心,省钱,快速模板网站建设找我们,业务覆盖抚州地区。费用合理售后完善,十年实体公司更值得信赖。

numpy.where(condition [,x,y])函数返回满足给定条件的输入数组中元素的索引。

参数:condition : When True, yieldx, otherwise yield y.x,y: Values from which to choose. x, y and

condition need to be broadcastable to some shape.

返回:out : [ndarray or tuple of ndarrays] If both x and y are specified, the output array contains elements of x where condition is True, and elements from y elsewhere.

If only condition is given, return the tuple condition.nonzero(), the indices where condition is True.

代码1:

# Python program explaining       # where() function         import numpy as np       np.where([[True, False], [True, True]],         [[1, 2], [3, 4]], [[5, 6], [7, 8]])

输出:

array([[1, 6],       [3, 4]])

代码2:

# Python program explaining       # where() function         import numpy as np       # a is an array of integers.     a = np.array([[1, 2, 3], [4, 5, 6]])       print(a)       print ('Indices of elements <4')       b = np.where(a<4)     print(b)       print("Elements which are <4")     print(a[b])

输出:

[[1 2 3] [4 5 6]] Indices of elements <4 (array([0, 0, 0], dtype=int64), array([0, 1, 2], dtype=int64)) Elements which are <4 array([1, 2, 3])

看完了这篇文章,相信你对Python中如何使用numpy.where()函数有了一定的了解,想了解更多相关知识,欢迎关注创新互联行业资讯频道,感谢各位的阅读!


网页名称:Python中如何使用numpy.where()函数
分享链接:http://pwwzsj.com/article/gggcgc.html