Operating An Array In Python Code