Operating An Array In Python Definition Function