Conditional reassignment of all array elements without for-loop