Array Operation In Python