Writing An Array In Python Interview