新聞中心
本文教程操作環(huán)境:windows7系統(tǒng)、Python 3.9.1,DELL G3電腦。

1、概念
numpy.where(condition [,x,y])函數(shù)返回滿足給定條件的輸入數(shù)組中元素的索引。
2、參數(shù)
condition
3、返回值
返回out
代碼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()函數(shù)的使用方法。
標題名稱:創(chuàng)新互聯(lián)Python教程:Python中numpy.where()函數(shù)的使用
當前URL:http://fisionsoft.com.cn/article/dhcepdp.html


咨詢
建站咨詢
