arithmetic operation on numpy array