Generate An Array In Python Interview