結果 : numpy array set values based on condition