我正在使用 np.where 函式來獲取值與 0 匹配的索引。但以下代碼回傳空陣列,這是不期望的。
import numpy as np
a = np.array([0,0,0,0])
np.where(a[a>=0])
這給出:(array([], dtype=int64),)。
誰能指出我缺少什么?
https://numpy.org/doc/stable/reference/generated/numpy.where.html
uj5u.com熱心網友回復:
這是預期的。a>=0給你所有True,即a[a>=0]給你a,它不包含任何非零元素。所以np.where(a)回傳空陣列。
你在找np.where(a>=0)嗎?
uj5u.com熱心網友回復:
你告訴找到索引,其中 a[ a>=0] a>=0 回傳 [True,True,True,True] 然后 a[a>=0] 回傳 [0,0,0,0] 然后沒有回傳的真實條件。
也許你想做
np.where(a>=0)
轉載請註明出處,本文鏈接:https://www.uj5u.com/yidong/335488.html
