Operating An Array In Python Problem