How to solve this KeyError:0 in JupyterLab?

Question

I run this code in Google Colab, it works well. But when I run it in my JupyterLab, this keyerror:0 happens. How can I solve this problem?

import sys
import sklea