Python Use Numpy Array Assignments