Update by an index and a mask? Update by an index and a mask? numpy numpy