Are you sure you didn't change any column of the original dataset? I simply tried to reproduce the problem, I altered very little in the code and I have a different output:
https://colab.research.google.com/drive/1myV1bPk9wBLAZeQ8tYKIocq-pFSLzVss?usp=sharing