結果 : numpy set array values based on condition