It's partially selection bias, yes.

But that's not the entire story. It's also the fact that not having to deal with the terrible kids helps the remaining kids. Fewer class disruptions. Less slowing down the class to pretend to let the slowest and least motivated keep up. Etc.

This comes at the cost of concentrating the troublemakers in other places, making them far worse for normal kids stuck there.