Finding Size Of Array In Python