天天看點

Numpy之where的用法1、numpy.where的傳回結果

1、numpy.where的傳回結果

numpy.where調用方式為numpy.where(condition,1,2)

滿足條件的位置上傳回結果1,不滿足的位置上傳回結果2

 例如通過where()函數将a數組中負值設為0,正值不變

Numpy之where的用法1、numpy.where的傳回結果

如果沒有指定傳回結果,隻有查找條件則傳回滿足條件的位置。傳回的結果是一個元組(tuple),包含兩個數組,第一個數組紀錄的是行,第二個數組紀錄的是列。

Numpy之where的用法1、numpy.where的傳回結果

可以使用zip函數将傳回的位置組成一個個坐标對,友善調用。zip函數直接傳回的是一個對象,可以用過for循環周遊出裡面的元素,也可以使用list直接列出所有坐标對元素。