Python While Loop For User Input Array Matlab Drive