Operating An Array In Python Problems