Restructuring Array In Python Interview