Code For Array In Python